检查pandas列是否包含列表中的所有元素

Question

我有这样的df：

frame = pd.DataFrame({'a' : ['a,b,c', 'a,c,f', 'b,d,f','a,z,c']})

以及物品清单：

letters = ['a','c']

我的目标是从中获取frame至少包含2个元素的所有行。letters

我想出了以下解决方案：

for i in letters:
    subframe = frame[frame['a'].str.contains(i)]

这给了我我想要的，但是就可伸缩性而言，它可能不是最佳的解决方案。是否有任何“向量化”解决方案？谢谢

Answer 1

我会建立一个Series列表，然后应用vectorized np.all：

contains = [frame['a'].str.contains(i) for i in letters]
resul = frame[np.all(contains, axis=0)]

它给出了预期的结果：

       a
0  a,b,c
1  a,c,f
3  a,z,c