python - 返回 pandas 数据框中的行，其中列中的元组包含特定值

Question

我正在尝试查询 pandas 数据框以查找其中一列包含包含特定值的元组的行。

举个例子:

   User                 Col1
0     1     (cat, dog, goat)
1     1         (cat, sheep)
2     1        (sheep, goat)
3     2          (cat, lion)
4     2  (fish, goat, lemur)
5     3           (cat, dog)
6     4          (dog, goat)
7     4                  cat

所以假设我想返回 Col1 包含“cat”的行，有没有一种方法可以做到这一点而无需遍历每一行并执行“if”(我的实际数据集有更多行)？

df['Col1'].isin(['cat'])

和

df['Col1'].str.contains("cat")

只为最后一行返回 'true'

Answer 1

您可以在 apply() 中使用 lambda 函数:

df[df["Col1"].apply(lambda x: "cat" in x)]

当 "cat" 在单元格中时，lambda 返回 True。这适用于字符串("cat"in "cat" is True)和元组("cat"in ("cat", "dog") 为 True)。通过对 df 进行子集化，您可以获得 lambda 为 True 的所有行。