检查Pandas中的元素（列表）中是否包含字符串

Question

我有一个pandas数据框，列为“朋友”。friends列中的所有元素都是list object。“朋友”列中的行示例：

['erik', 'jason', 'eduard']

我想返回一个dataframe，在其中我检查eduard是否是列表。这是我的示例代码：

^{pr2}$

请帮一只猫出去

Answer 1

可以使用集合交集：

In [66]: df
Out[66]:
                 friends
0  [erik, jason, eduard]
1            [anna, max]
2       [eduard, natali]

In [67]: df[df.friends.map(set) & {'anna','natali'}]
Out[67]:
            friends
1       [anna, max]
2  [eduard, natali]

Answer 2

您需要apply在这种情况下，使用普通的in运算符来检查列表是否包含eduard：

df[df.friends.apply(lambda x: 'eduard' in x)]

示例：

^{pr2}$