python - 通过与列表比较过滤掉 panda-df 的行

Question

这个问题已经有答案了:

How do I select rows from a DataFrame based on column values? (17 个回答)

How to filter Pandas dataframe using 'in' and 'not in' like in SQL (12 个回答)

已关闭 5 年前。

经常出现以下问题。假设我有一个数据框，其中一列可以采用离散值:

df = pd.DataFrame({'col1': [1, 2,3,4,5,6,7], 'col2': ["A", "B", "A", "C", "B", "A", "D"]})

在本例中col2可以取值 A、B 或 C。我只想要 col2 所在的行不等于 A 或 B。我认为以下语法可行，

df["col2"] not in ["A", "B"]

但是，这给了我错误 ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

有没有一种巧妙的方法来过滤掉这些行？

Answer 1

您可以使用isin方法。

df = df[~df.col2.isin(['A', 'B'])]

输出

   col1 col2
3     4    C
6     7    D