过滤条件 pandas df 包含一个列表

Posted 2023-03-12

【中文标题】过滤条件 pandas df 包含一个列表

【英文标题】：Filtering with conditions pandas df containing a list

【发布时间】：2021-10-26 17:52:35

【问题描述】：

我有一个 df，它的单元格中有列表对象：

data['country_code']
0       [IT, IT]
1       [PL, PL]
2       [IT, IT]
3       [IT, IT]
4       [IT, IT]
          ...   
6318    [XX, MT]
6319    [FI, FI]
6320    [XX, XX]
6321    [FI, FI]
6322    [FI, FI]
Name: country_code, Length: 6323, dtype: object

如果data['country_code'] 中的列表将'SK' 或'CZ' 作为第一个或第二个元素，我想过滤数据框data

类似这样的：

data[first element of data['country_code'] == 'SK'or'CZ' or second element of data['country_code'] == 'SK'or'CZ']

在 MongoDB 语法中是：

.find($or: [country_code: $elemMatch = 'SK', country_code: $elemMatch = 'CZ'])

【参考方案1】：

你可以使用：

print(df[df.country_code.apply(lambda x: "SK" in x or "CZ" in x)])

打印：

  country_code
3     [IT, CZ]
4     [SK, IT]

---

df 已使用：

  country_code
0     [IT, IT]
1     [PL, PL]
2     [IT, IT]
3     [IT, CZ]
4     [SK, IT]

【讨论】：

工作，你！将在 5 分钟内接受它。顺便说一句，感觉有点像鸡编码。但是它有效！

【参考方案2】：

你也可以使用pd.Series.str访问器：

l = ['SK', 'CZ']

print (data[data['country_code'].str[0].isin(l)|data["country"].str[1].isin(l)])