如何过滤包含特定项目的列表的 DataFrame 列

Question

如果我想为包含特定术语的字符串过滤一列字符串，我可以这样做：

df = pd.DataFrame({'col':['ab','ac','abc']})
df[df['col'].str.contains('b')]

返回：

   col
0   ab
2  abc

如何过滤包含特定项目的列表列？例如，来自

df = pd.DataFrame({'col':[['a','b'],['a','c'],['a','b','c']]})

如何获取所有包含“b”的列表？

         col
0     [a, b]
2  [a, b, c]

Answer 1

你可以像这样使用apply。

In [13]: df[df['col'].apply(lambda x: 'b' in x)]
Out[13]: 
         col
0     [a, b]
2  [a, b, c]

虽然一般来说，将列表存储在 DataFrame 中有点尴尬 - 您可能会发现一些更易于使用的不同表示（列表中每个元素的列、MultiIndex 等）。