Как выбрать указанный c элемент строки для построения диаграммы в python

Question

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

Я использую этот код и его график для каждой страны, которую я имею в CSV, как я могу выбрать только некоторые страны, такие как Китай?

Answer 1

Если хотите отфильтровать только один фильтр страны по Series.eq и boolean indexing, то groupby не требуется:

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

При необходимости Для фильтрации нескольких стран из списка используйте boolean indexing с Series.isin:

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

Answer 2

Определите список стран, отфильтруйте ваш фрейм данных и график

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