从创建的列表中过滤数据

Question

我正在处理来自 github 的 Covid 数据集，我想用 csv 格式的 EU_member list 中出现的国家/地区过滤我的数据集。

import pandas as pd

df = pd.read_csv('https://raw.githubusercontent.com/owid/covid-19-data/master/public/data/owid-covid-data.csv')
df = df[df.continent == 'Europe']

# From here I want to  just pick those countries that appear in the following list:
EU_members= ['Austria','Italy','Belgium''Latvia','Bulgaria','Lithuania','Croatia','Luxembourg','Cyprus','Malta','Czechia','Netherlands','Denmark','Poland','Estonia',
'Portugal','Finland','Romania','France','Slovakia','Germany','Slovenia','Greece','Spain','Hungary','Sweden','Ireland']

# I have tried something like this but it is not what I expected:
df.location.str.find('EU_members')

Answer 1

您可以使用 .isin():

import pandas as pd

df = pd.read_csv('https://raw.githubusercontent.com/owid/covid-19-data/master/public/data/owid-covid-data.csv')

EU_members= ['Austria','Italy','Belgium''Latvia','Bulgaria','Lithuania','Croatia','Luxembourg','Cyprus','Malta','Czechia','Netherlands','Denmark','Poland','Estonia',
'Portugal','Finland','Romania','France','Slovakia','Germany','Slovenia','Greece','Spain','Hungary','Sweden','Ireland']


df_out = df[df['location'].isin(EU_members)]
df_out.to_csv('data.csv')

创建 data.csv:

从创建的列表中过滤数据

Filter data from a created list

python

filtering

pandas