python - Pandas Where方法: How to query if a value is in a list that is a value

Question

我有下面的数据框:

           a
0  [1, 2, 3]
1  [5, 6, 7]

我正在尝试查找 b 列(列表)的值包含 1 的行。因此，对于上面的表，查询将返回:

           a
0  [1, 2, 3]

我尝试过:

> df.where(1 in df.a)
ValueError: where requires an ndarray like object for its condition

和:

> df.query('1 in a')
KeyError: True

有人可以告诉我是否有办法使用 where 或查询方法来执行此操作吗？

谢谢

Answer 1

可能有更好的方法，但一个简单的方法是使用 apply 和 lambda 函数访问列表:

df[df.a.apply(lambda seq: 1 in seq)]