# Initial import. Load code to the console
from plotSlope import slope
import pandas as pd
import os


# Load data from a csv file using pandas and display it 
data = pd.read_csv(os.path.join('Data','EU_GDP_2007_2013.csv'),index_col=0,na_values='-')
data.head()/1000

# Specify an optional dictionnary of color 
color  = {"France":'blue',
          'Germany':'red',
          'Ireland':'chocolate',
          'United Kingdom': 'purple', 
          'Poland':'green'}
color

f = slope(data/1000,kind='interval',color=color,
          height= 18,width=20,font_size=15,font_family='GillSans',dpi=150,savename='test.png',
          title = u'European GPD until 2010 and forecasts at market prices (billions of Euro) source : EUROSTAT') 

# Changing font and figure size.
# Notice how some label get lumped together
f = slope(data/1000,width =30,height= 12,kind='interval',font_size=20,marker='%0.f',color=None,savename=None,dpi=200) 

# Without numbers
f = slope(data/1000,width =15,height= 12,kind='interval',marker = None,font_size=12,color=color,savename=None,dpi=200) 

cancer_data = pd.read_csv(os.path.join('Data','cancer_survival_rate.csv'),index_col=0)
cancer_data.head()

f = slope(cancer_data,height= 18,width=18,font_size=20,savename='cancer.png')   

cancer_color = {'Prostate':'chocolate','Leukemia':'red'}

f = slope(cancer_data,height= 18,width=18,font_size=20,color=cancer_color,savename='cancer.png')