python - 如何过滤具有列表的列的数据框包含值

<分区>

我们的数据框在一列中包含列表。找不到简单的方法来过滤列表中包含值的行的数据框。

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

例如，我只需要在列表中包含“a”的行。我设法仅通过“申请”获得了它。

df[df.lists.apply(lambda x: True if 'a' in x else False)]
>>>     lists
>>>0    [a, c]
>>>1    [a, b, d]

是否有类似 .isin() 的东西，反之亦然？获取所需行的最佳方法是什么？谢谢。

最佳答案

最简单的是使用 apply 和 in:

df1 = df[df.lists.apply(lambda x: 'a' in x)]

但是如果要检查a创建DataFrame，但是有点复杂:

df1 = df[pd.DataFrame(df.lists.values.tolist()).eq('a').any(axis=1)]

另一个解决方案是使用 str.join与 str.contains :

df1 = df[df.lists.str.join(',').str.contains('a')]

print (df1)
       lists
0     [a, c]
1  [a, b, d]

关于python - 如何过滤具有列表的列的数据框包含值，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/49008508/