Я пытаюсь создать программу Python, которая будет показывать следующие данные о Covid-19, все из файла CSV:
- случаев в день
- всего случаи
- смертей в день
- всего смертей
Я читаю данные из файла и заполняю информацию из 4 массивов. Я могу перебрать массивы и распечатать данные - все выглядит хорошо.
Проблема в том, что когда я смотрю на полученный график, данные кажутся нормальными для первых 25 записей, а затем увеличиваются со всеми линии параллельно. Шкала на оси Y также неверна - похоже, она перезапускает шкалу для каждого из 4 наборов данных.
Вот код:
import csv
import matplotlib.pyplot as plt
from datetime import datetime
#create the arrays to hold virus data
caseDate, cases, casesCum, deaths, deathsCum = [], [], [], [], []
leg1 = "Cases"
leg2 = "Cases - Cum"
leg3 = "Deaths"
leg4 = "Deaths - Cum"
#open virus data file
filename = 'Virus Data.csv'
with open(filename) as f:
reader = csv.reader(f)
header_row = next(reader)
#record data in arrays
for row in reader:
current_date = datetime.strptime(row[0], "%d/%m/%Y")
caseDate.append(current_date)
cases.append(row[1])
casesCum.append(row[2])
deaths.append(row[3])
deathsCum.append(row[4])
for i in range (len(cases)):
print("Entry ", i, ": Cases: ", cases[i],"; Cum Cases: ", casesCum[i],"; Deaths: ", deaths[i],"; Cum Deaths: ", deathsCum[i])
#data now recorded
#plot data in chart
fig = plt.figure(dpi = 128, figsize=(12,6))
plt.plot(caseDate, cases, c="red", alpha=0.5)
plt.plot(caseDate, casesCum, c="blue", alpha=0.5)
plt.plot(caseDate, deaths, c="green", alpha = 0.5)
plt.plot(caseDate, deathsCum, c="black", alpha = 0.5)
#format plot
title = "Covid-19 Statistics"
plt.title(title, fontsize = 20)
plt.xlabel("", fontsize = 16)
fig.autofmt_xdate()
plt.ylabel("Cases / Cum Cases / Deaths / Cum Deaths", fontsize = 12)
plt.tick_params(axis='both', which='major', labelsize=16)
plt.show()
Это почти там, но не совсем. Когда я создаю диаграмму в Excel, она выглядит так, как я ожидал:
Я бы хотел, чтобы она работала в Python .... .