python - Pandas :检查是否存在具有某些值的行

Question

我有一个像这样的二维(或更多)pandas DataFrame:

>>> import pandas as pd
>>> df = pd.DataFrame([[0,1],[2,3],[4,5]], columns=['A', 'B'])
>>> df
   A  B
0  0  1
1  2  3
2  4  5

现在假设我有一个像 np.array([2,3]) 这样的 numpy 数组，并且想检查 df 中是否有任何行与我的数组的内容。在这里，答案显然应该是正确的，但是例如。 np.array([1,2]) 应该返回 false，因为在 A 列中不存在同时具有 1 和 B 列中的 2 的行。

当然这很简单，但现在看不到。

Answer 1

事实证明这真的很容易，以下是这里的工作:

>>> ((df['A'] == 2) & (df['B'] == 3)).any()
True
>>> ((df['A'] == 1) & (df['B'] == 2)).any()
False

也许有人想出了一个更好的解决方案，它允许直接传入数组和要匹配的列列表。

请注意，df['A'] == 2 周围的括号不是可选的，因为 & 运算符的绑定(bind)与 ==< 一样强 运算符。