In [86]:

    

import pandas as pd
from pandas import io

import matplotlib.pyplot as plt

%matplotlib inline


    

In [87]:

    

df = pd.read_stata('data/trade/trade_surpluses.dta')


    

In [88]:

    

df = df[['year', 'countrya', 'surplus_all']]
df = df[df['year'] >= 1972]


    

In [89]:

    

# export - import
trade_balance = df.groupby(['year', 'countrya'])['surplus_all'].first().unstack('year').sort_index()


    

In [90]:

    

trade_balance.to_csv('data/raw_imputed/trade_balance_new.csv')