In [58]:

    

import numpy as np
import pandas as pd
import csv
import matplotlib as mpl
import matplotlib.pyplot as plt
from matplotlib import


    

    




  File "<ipython-input-58-efc97081b61e>", line 6
    from matplotlib import
                           ^
SyntaxError: invalid syntax

In [56]:

    

df = pd.read_csv('test_2.csv')

#print df['shock'][0]=='-10%'



#sns.pointplot(x="current_step", y="system_TAS", hue="shock", data=df,
#              palette={"-10%": "g", "-20%": "m", "-30%": "c"},
#              markers=["^", "o", "x"], linestyles=["-", "--", "---"]);

keys = np.unique(df['shock'])



for key in keys:
    I = np.argwhere(df['shock']==key).squeeze()

#print df['current_step'][I]

    plt.figure('fig 1')
    plt.plot(df['current_step'][I], df['system_TAS'][I], label="system_TAS %s"%key)
    plt.xlim(0, 7)
    plt.xlabel("period")
    plt.ylim(-1e12, 0.2e12)
    plt.ylabel(r"Asset $\delta$ sales")
    plt.title("My title is")


#    plt.figure('fig 2')
#    plt.plot(np.linspace(0,1), np.linspace(0,1))

plt.legend()


    

    
Out[56]:





<matplotlib.legend.Legend at 0x110e1a450>

In [47]:

    




    

In [53]: