如何选择特定的行元素以在python中绘制图表

Question

df.set_index('day', inplace=True)
df.groupby('Country Name')['Cumulative Confirmed'].plot(legend=True)

我使用此代码，并从CSV中的每个国家/地区绘制图表，如何仅选择某些国家/地区，例如中国？

Answer 1

定义国家/地区列表，过滤数据框并作图

country_list = ['CountryA', 'CountryB', 'CountryC']
df.query('`Country Name` in @country_list')\
    .groupby('Country Name')['Cumulative Confirmed'].plot(legend=True)

Answer 2

如果只过滤一个国家：

df.loc[df['Country Name'].eq('China'), 'Cumulative Confirmed'].plot(legend=True)

如果需要过滤多个国家：

(df[df['Country Name'].isin(countries)]
      .groupby('Country Name')['Cumulative Confirmed'].plot(legend=True))