【Python】进阶学习：pandas--isin()用法详解

isin() 是 pandas 中的一个方法，用于检查一个 Series 或 DataFrame 中的元素是否在传递的值序列中。它返回一个布尔 Series 或 DataFrame，表示每个元素是否在指定的值序列中。

使用方法

1. 对于一个 Series 使用 isin()

import pandas as pd
 
# 创建一个 Series
s = pd.Series(['a', 'b', 'c', 'd'])
 
# 使用 isin() 方法
print(s.isin(['a', 'b']))

2. 对于一个 DataFrame 使用 isin()

import pandas as pd
 
# 创建一个 DataFrame
df = pd.DataFrame({
    'A': ['a', 'b', 'c', 'd'],
    'B': ['e', 'f', 'g', 'h']
})
 
# 使用 isin() 方法
print(df.isin({'A': ['a', 'b'], 'B': ['e', 'f']}))

3. 使用 ~ 来获取不在指定值序列中的元素

import pandas as pd
 
# 创建一个 Series
s = pd.Series(['a', 'b', 'c', 'd'])
 
# 使用 isin() 方法获取不在指定值序列中的元素
print(s[~s.isin(['a', 'b'])])

注意

• isin() 方法对于 NaN 值总是返回 False。
• 传递给 isin() 的值序列可以是列表、元组、集合、Series 或者任何可迭代对象。