使用任何列表值筛选数据[duplicate]

Question

我有一个数据框：

df
0       PL
1       PL
2       PL
3       IT
4       IT
        ..
4670    DE
4671    NO
4672    MT
4673    FI
4674    XX
Name: country_code, Length: 4675, dtype: object

我通过以下方式按德国国家标签“DE”过滤此内容：

df = df[df.apply(lambda x: 'DE' in x)]

如果我想筛选更多的国家，我必须通过以下方式手动添加它们：。应用（lambda x：'DE'in x或'GB'in x）。然而，我想创建一个国家列表，并自动生成此声明。

大概是这样的：

countries = ['DE', 'GB', 'IT']
df = df[df.apply(lambda x: any_item_in_countries_list in x)]

我想我可以过滤df 3次，然后通过concat（）将这些片段合并回来，但是有没有更通用的函数来实现这一点？

Answer 1

如果你有列名，你可以试试这个

countries = ['DE', 'GB', 'IT']
df[df['country_code'].isin(countries)]

Answer 2

您可以使用.isin（）：

df[df['country_code'].isin(['DE', 'GB', 'IT'])]

性能比较：

import timeit
import pandas as pd
df = pd.DataFrame({'country_code': ['DE', 'GB', 'IT', 'MT', 'FI', 'XX'] * 1000})

%timeit df[df['country_code'].isin(['DE', 'GB', 'IT'])]
409 µs ± 19 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

%timeit df['country_code'].apply(lambda x: x in ['DE', 'AT', 'GB'])
1.35 ms ± 474 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)