python - 如何对列值为列表的 pandas DataFrame 进行子集化？

Question

我有一个 pandas DataFrame df 看起来像这样:

A      B
0  ['a','b']
1  ['c','d']
2  ['a','c']
3  ['b','d']
4  ['a','d']

现在，我希望通过仅选择 'a' 属于 B 列表的那些行来对 df 进行子集化，所需的输出为:

A      B
0  ['a','b']
2  ['a','c'] 
4  ['a','d']

天真地，我尝试了 df['a' in df['B']]，但这似乎不起作用。我该怎么做呢？

Answer 1

使用 apply 是一种过滤方法。

In [39]: df[df['B'].apply(lambda x: 'a' in x)]
Out[39]:
   A       B
0  0  [a, b]
2  2  [a, c]
4  4  [a, d]