如何应用函数可以用python-pandas测试数据框中列表中的元素是否存在?

时间:2016-10-06 15:51:08

标签: python pandas

对于数据框df,其列col包含list值:

 id    col
 1     [1, 10, 23]
 2     [2, 11, 19, 29]
 ..

我试过了:

df[1 in df.col]

但得到了一个错误:

KeyError: True

你知道我怎样才能恰当地实现它?提前谢谢!

2 个答案:

答案 0 :(得分:2)

选项使用apply
df.col.apply(lambda x: 1 in x)

<强> 演示
df[df.col.apply(lambda x: 1 in x)]
enter image description here

答案 1 :(得分:1)

尝试阅读pandas document中的“apply”功能。

df['has_element'] = df['col'].apply(lambda x: element in x)