使用matplotlib绘制多个列表时出现问题

时间:2020-05-03 16:36:42

标签: python matplotlib data-visualization

我正在编写一个脚本,该脚本可用于绘制按国家/地区划分的covid时间序列数据。当我绘制一个国家/地区时,它工作正常,但在Y轴上的比例尺已正确打印。 Plot which I am getting问题是在打印了一个国家/地区的最大值之后,使用较小的值外推y轴以绘制后续国家/地区的数据点。 我的脚本的代码如下

import requests
from contextlib import closing
import csv
import matplotlib.pyplot as plt
url = "https://raw.githubusercontent.com/datasets/covid-19/master/data/countries-aggregated.csv"

def prepareCountryWiseData(country):
    countryWise = {}
    with closing(requests.get(url, stream=True)) as r:
        f = (line.decode('utf-8') for line in r.iter_lines())
        reader = csv.reader(f, delimiter=',', quotechar='"')
        active = []
        recovered = []
        dates = []
        for row in reader:    
             if row[1] == country:
                     dates.append(row[0])
                     active.append(row[2])
                     recovered.append(row[3])
        return (dates, active, recovered)

def plotCountryWiseData(countryList):
    plotable = []
    for country in countryList:
            dates,active,recovered = (prepareCountryWiseData(country))
            plt.plot(active)                
    plt.ylabel('active_cases')
    plt.legend(countryList)
    plt.show()
    plotCountryWiseData(['India','US','Italy'])

1 个答案:

答案 0 :(得分:1)

如果您可以使用pandas模块,那么您的工作将会容易得多:

import pandas as pd, matplotlib.pyplot as plt

url = "https://raw.githubusercontent.com/datasets/covid-19/master/data/countries-aggregated.csv"
df = pd.read_csv(url)

fig,ax = plt.subplots()

for k,g in df[df['Country'].isin(['India','US','Italy'])].groupby('Country'):
    ax = g.plot(ax=ax,kind='line',x='Date',y='Confirmed',label=k) 

plt.gcf().suptitle('Active Cases')
plt.show()

结果:
enter image description here