python DataFrame isin() 与 ~isin()

文章目录

  • 一、生成测试数据集
  • 二、isin()
    • 2.1 单一条件筛选
    • 2.2 多个条件筛选
  • 三、~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))],
    }
)

python DataFrame isin() 与 ~isin()_第1张图片

二、isin()

2.1 单一条件筛选

  • 筛选 name 包含[‘云韵’,‘美杜莎’] 的数据
filt = df['name'].isin(['云韵','美杜莎'])
df[filt]

python DataFrame isin() 与 ~isin()_第2张图片

2.2 多个条件筛选

  • 筛选 name 包含[‘云韵’,‘美杜莎’] 或 最小age
filt_1 = df['name'].isin(['云韵','美杜莎'])
filt_2 = df['age'] == df['age'].min()
df[filt_1 | filt_2]

python DataFrame isin() 与 ~isin()_第3张图片

三、~isin()

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

python DataFrame isin() 与 ~isin()_第4张图片

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

你可能感兴趣的:(python,数据分析)