In [1]:
%matplotlib notebook
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
In [2]:
data=pd.read_csv("05_temp_rain_v2.csv")
#data.columns
#data
In [5]:
pivot=data.pivot_table(["rain(mm)","temp(dC)"], index=["loc","month"])#.plot()
pivot.head()
Out[5]:
In [22]:
pivot=data.pivot_table(["rain(mm)","temp(dC)"], index=["loc","month"])#.plot()
pivot
#pivot.xs("Agna", level=0)
locations=pivot.index.get_level_values(0).unique()
print(locations)
months=pivot.index.get_level_values(1).unique()
print(months)
for location in locations:
#print(pivot.xs(location, level=0))
split=pivot.xs(location)
print(split)
#split.plot()
#break
rain=list(split["rain(mm)"])#.plot(kind="bar")
temp=split(["temp(dC)"])#.plot()
print("printing rain")
print(list(rain))
print("printing temp")
print(list(temp))
#rain.plot(kind="bar")
#temp.plot()
i = np.arange(len(rain))
fig, rr = plt.subplots()
#bar_width = 1
opacity = 0.4
#error_config = {'ecolor': '0.3'}
rr.bar(i, rain, alpha=opacity, color='b',
align='center' )
tt=plt.twinx()
tt.plot(temp, linestyle='-', color="r", linewidth=1.0, label="temp")
plt.xlabel('months')
rr.set_ylabel('(mm)')
#plt.ylabel('(mm)')
tt.set_ylabel('(C)')
text="rain and temp of " + location
plt.title(text)
plt.xticks(i,('jan', 'feb', 'mar', 'apr', 'mai', 'jun', 'jul', 'aug', 'sept', 'oct', 'nov', 'dez'))
plt.legend()
plt.grid(True)
#rr.set_ylim([0,40])
plt.tight_layout()
plt.show()
break
#rain=pivot.xs(location, level=0)["rain(mm)"]
#temp=pivot.xs(location, level=0)["temp(dC)"]
#print(rain)
#rain.plot(kind="bar")
#temp.plot()
#plt.show()