详解 Pandas 的 isin 用法

Pandas 的 isin() 方法可以判断数据值是否在某个数据集合中，若与集合中的某个值相等则返回 True，反之返回 False。

import pandas as pd

df = pd.DataFrame({
    "title": ["one", "two", "three", "four"],
    "type": ["small", "common", "middle", "large"],
    "num": [10, 20, 30, 40]
})

# 1.判断某列的值是否在指定集合中
print(df["title"].isin(["one", "five"]))

结果展示

0     True
1    False
2    False
3    False

# 2.判断所有列的值是否在指定集合中
print(df.isin(["two", "large", 30]))

结果展示

  title   type    num
0  False  False  False
1   True  False  False
2  False  False   True
3  False   True  False

# 3.判断不同的列是否在不同的指定集合中
print(df.isin({
    "title": ["one", "four"],
    "type": ["common", "middle"],
    "num": [10, 20]
}))

结果展示

   title   type    num
0   True  False   True
1  False   True   True
2  False   True  False
3   True  False  False