46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
import json
|
|
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
|
|
colorList = json.load(open('color/config.json','r'))["color"]
|
|
|
|
import csv
|
|
|
|
with open('data/passenger.csv', 'r') as f:
|
|
reader = csv.reader(f)
|
|
header = next(reader)
|
|
data = [row for row in reader]
|
|
|
|
data = np.array(data).astype(int).T
|
|
|
|
bar_width = 0.25
|
|
for i in range(1,len(data)):
|
|
plt.bar(data[0]+(i-2)*bar_width, data[i], label=header[i], color=colorList[i-1], width=bar_width)
|
|
|
|
ax2 = plt.twinx()
|
|
ax2.plot(data[0],data[2]/data[1],label="income/pop",color=colorList[len(data)-1])
|
|
|
|
# with open('data/temperature.csv', 'r') as f:
|
|
# reader = csv.reader(f)
|
|
# header = next(reader)
|
|
# data = [row for row in reader]
|
|
|
|
plt.legend()
|
|
|
|
# data = np.array(data).astype(float).T
|
|
# ax2 = plt.twinx()
|
|
# for i in range(1,len(data)):
|
|
# if i <=2 :
|
|
# # data[i] = ( data[i] - 32 ) / 1.8
|
|
# # ax2.plot(data[0], data[i], label=header[i], color=colorList[i-1+len(data)-1])
|
|
# pass
|
|
# else:
|
|
# ax2.plot(data[0], data[i], label=header[i], color=colorList[i-1+len(data)-1])
|
|
|
|
# ax2.set_ylabel('Temperature (Celcius)')
|
|
# # ax2.set_ylim(-10,20)
|
|
# ax2.legend(loc='upper left')
|
|
|
|
plt.xlabel('Year')
|
|
|
|
plt.show() |