This repository has been archived on 2025-02-02. You can view files and clone it, but cannot push or open issues or pull requests.
Files_for_MM/passenger-and-revenue-simple-plot.py
2025-01-26 17:26:49 +08:00

34 lines
1020 B
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(float).T
data[0]=data[0].astype(int)
bar_width = 0.35
plt.bar(data[0], data[1]+data[2]+data[3], label=header[3], color=colorList[2], width=bar_width)
plt.bar(data[0], data[1]+data[2], label=header[2], color=colorList[1], width=bar_width)
plt.bar(data[0], data[1], label=header[1], color=colorList[0], width=bar_width)
# plt.plot(data[0],data[4], label=header[4], color=colorList[3])
plt.legend()
# plt.xlabel('Year')
plt.ylabel('Passenger')
plt.xticks(data[0],rotation=45)
ax2=plt.twinx()
ax2.plot(data[0],data[5], label=header[5], color=colorList[4])
ax2.set_ylabel('Revenue')
plt.legend(loc='upper right')
plt.title('Passenger and Revenue Yearly')
plt.savefig('result/passenger-and-revenue.png',dpi=1024)
plt.show()