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()