pandas - 选择可以在列表中找到数字的行

Question

给定以下数据

我希望选择list中出现num的行。在这种情况下，它将选择第 1 行和第 2 行，第 3 行未被选择，因为在 [4,5] 中找不到 3。

下面是dataframe，我们应该怎么写filter query呢？

cat1=pd.DataFrame({"num":[1,2,3],
                 "list":[[1,2,3],[3,2],[4,5]]})

Answer 1

一个可能的解决方案是列表理解，zip 和 in 传递给 boolean indexing :

df = cat1[[a in b for a, b in zip(cat1.num, cat1.list)]]

或使用 DataFrame.apply 的解决方案使用 axis=1 进行每行处理:

df = cat1[cat1.apply(lambda x: x.num in x.list, axis=1)]

或者创建DataFrame并测试成员资格:

df = cat1[pd.DataFrame(cat1.list.tolist()).isin(cat1.num).any(axis=1)]

---

print (df)
   num       list
0    1  [1, 2, 3]
1    2     [3, 2]