Python как создать новый набор данных из существующего на основе условия

Question

Например: у меня есть этот код:

import pandas
df = pandas.read_csv('covid_19_data.csv')

в этом наборе данных есть столбец с именем countryterritoryCode, который является кодом страны. образец данных из набора данных

В этом наборе данных содержится информация о случаях заболевания covid19 во всех странах мира. Как создать новый набор данных, в котором отображается только информация о США (где countryterritoryCode == USA)

Answer 1

import pandas
df = pandas.read_csv('covid_19_data.csv')
new_df = df[df["country"] == "USA"]
or
new_df = df[df.country == "USA"]

Answer 2

Используйте df.groupby:

df = pandas.read_csv('covid_19_data.csv')

df_new = df.groupby('countryterritoryCode', axis = 1)