In [2]:

    

%matplotlib notebook
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
#import seaborn as sns


    

In [16]:

    

data=pd.read_csv("05_temp_rain_v2.csv")
pivot=data.pivot_table(["rain(mm)","temp(dC)"], ["loc","month"])
#print(pivot)
locations=pivot.index.get_level_values(0).unique()
months=pivot.index.get_level_values(1).unique()



for location in locations:
    split=pivot.xs(location).reset_index()
    #print(split)
    rain=split["rain(mm)"]
    temp=split["temp(dC)"]
    
    fig = plt.figure()
    ax1 = fig.add_subplot(111)
    ax1.plot(temp, color="r")
    ax1.set_ylabel("temp")
    #ax1.set_xlim(0,13)
    ax1.set_ylim(-20,50)
    ax1.spines['bottom'].set_position(("data",0))
    
    
    labels = ['Jan','Feb','Mar','Apr','May','Jun', 'Jul','Aug','Sep','Oct','Nov','Dez']
    plt.xticks(range(12),labels,rotation=45)
    ax1.set_xticklabels(labels, rotation=45)
    
    ax2 = ax1.twinx()
    ax2.plot(rain)
    ax2.set_ylabel("rain")
    ax2.set_ylim(-20,500)
    ax2.spines['bottom'].set_position(("data",0))

    
    plt.grid(True)
    plt.tight_layout()
    #plt.axes().set_aspect("equal")