Python | 熊猫 Index.contains()

简介

在 Pandas 中，Index.contains() 方法用于检查索引是否包含特定的值或多个值。返回布尔值数组，其中包含每个值是否在索引中出现。

语法

Index.contains(value, *, level=None)

参数

• value：要检查的值或值的列表或值的数组。
• level ：如果索引是 MultiIndex 且要检查的值位于一个级别中，则一定要给 level 传入相应的 index level值。

返回值

返回一个布尔值数组，其中包含每个传入的值是否在索引中。

示例

# 导入 Pandas 包
import pandas as pd

# 创建一个示例 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df = df.set_index('A')

# 使用 Index.contains() 方法检查索引是否包含特定的值
print(df.index.contains(1))  # 输出 True
print(df.index.contains([1, 4]))  # 输出 [True, False]

输出：

True
[ True False]

结论

Index.contains() 方法是一个非常有用的方法，用于检查 Pandas DataFrame 或 Series 的索引是否包含特定的值。它可以帮助我们快速确定是否存在某个值，从而进行相应的操作。