我有一个熊猫数据框和“朋友”列。friends列中的所有元素都是list对象“朋友”列中的一行示例:
['erik', 'jason', 'eduard']
我想返回一个数据框,在其中我检查eduard是否是列表。这是我的示例代码:
df[df.friends.isin(['eduard'])
请帮一只猫
最佳答案
在这种情况下,您需要使用normalapply
运算符检查列表是否包含in
:
df[df.friends.apply(lambda x: 'eduard' in x)]
例子:
df = pd.DataFrame({"friends": [['erik', 'jason', 'eduard'], ['erik', 'jason']]})
df[df.friends.apply(lambda x: 'eduard' in x)]
# friends
#0 [erik, jason, eduard]