Pandas 检查行是否存在某些特定值

在本文中，我们将介绍如何在 Pandas 中检查特定值存在于 DataFrame 中的行。

准备工作

首先，我们需要导入 Pandas 包并创建一个 DataFrame 示例：

import pandas as pd

data = {
  'name': ['Alice', 'Bob', 'Charlie', 'David'],
  'age': [25, 30, 35, 40],
  'favorite_color': ['blue', 'green', 'red', 'yellow']
}

df = pd.DataFrame(data)

现在我们有一个包含姓名、年龄和喜欢的颜色的 DataFrame。让我们通过使用 head() 函数来查看前几行：

     name  age favorite_color
0   Alice   25           blue
1     Bob   30          green
2  Charlie   35            red
3    David   40         yellow

检查行是否存在某些特定值

假设我们想检查 DataFrame 中是否存在姓名为 Bob，年龄为 30，喜欢的颜色为绿色的行。我们可以使用布尔索引来完成。布尔索引可以根据条件筛选 DataFrame 的行。下面的代码演示了如何使用布尔索引进行检查：

mask = (df['name'] == 'Bob') & (df['age'] == 30) & (df['favorite_color'] == 'green')

if df[mask].empty:
  print('Row does not exist')
else:
  print('Row exists')

在此示例中，我们首先创建了一个名为 mask 的变量。这个变量使用逻辑运算符 &，将条件连接起来，并创建一个布尔变量来指示哪些行匹配条件。

接下来，我们使用 mask 变量来选择 DataFrame 中的行。如果找不到符合条件的行，我们将打印消息“Row does not exist”，反之则打印消息“Row exists”。

以上代码的输出将是：“Row exists”。