python DataFrame isin() 和 ~isin()

一、生成测试数据集

import random
import pandas as pd

names = ['南宫婉','雅菲','余靖秋','姜立','小舞','胡列娜','比比东','云韵','美杜莎','焰灵姬']
df = pd.DataFrame(
    {
        "name"  :names,
        "age"   :[random.randint(18,25) for i in range(len(names))],
        "height":[round(random.uniform(1.6,1.75),2) for i in range(len(names))],
        "weight":[random.randint(45,50) for i in range(len(names))],
    }
)

二、isin()

2.1 单一条件筛选

• 筛选 name 包含[‘云韵’,‘美杜莎’] 的数据

filt = df['name'].isin(['云韵','美杜莎'])
df[filt]

2.2 多个条件筛选

• 筛选 name 包含[‘云韵’,‘美杜莎’] 或 最小age

filt_1 = df['name'].isin(['云韵','美杜莎'])
filt_2 = df['age'] == df['age'].min()
df[filt_1 | filt_2]

三、~isin()

• 结果与isin()相反
• 筛选 name 不包含[‘云韵’,‘美杜莎’] 的数据

filt = ~df['name'].isin(['云韵','美杜莎'])
df[filt]

• 参考：Python Pandas DataFrame.isin()用法及代码示例

这篇好文章是转载于：学新通技术网