python - 从 pandas 数据框中有效地提取行，我必须在其中搜索列表中的元素

Question

我的数据框是这样的:

id author
1  ['abc','def','alp']
2  ['akx','xyz','shdj','dhhs']
3  ['abc', 'alp','pqw']

例如，我想提取作者为 ['abc','alp'] 的所有行。它应该提取第 1 行和第 3 行。

有没有什么有效的方法可以使用 python pandas 来做到这一点？

Answer 1

使用带有列表理解的集合，用于过滤 boolean indexing :

L = ['abc','alp']
df = df[[set(x) >= set(L) for x in df.author]]
print (df)
   id           author
0   1  [abc, def, alp]
2   3  [abc, alp, pqw]

详细信息:

print ([set(x) >= set(L) for x in df.author])
[True, False, True]