过滤熊猫上的列表元素

Question

我有一个熊猫数据框foo，如下所示：

   a         b
0  1    [1, 2]
1  2    [1, 2]
2  3  [10, 11]

，并且我试图检索列[1, 2]中包含元素b的所有行。有什么好方法吗？

Answer 1

执行此操作的标准方法是使用boolean indexing：

mask = df['b'].apply(lambda x: x == [1, 2])
df[mask]

返回

   a       b
0  1  [1, 2]
1  2  [1, 2]

如果您不熟悉熊猫，new user tutorial是一个很好的起点，它将涵盖诸如此类的问题。