跳转至

🔥AI副业赚钱星球

点击下面图片查看

郭震AI

技巧23: 筛选出满足要求的分组

输入

import pandas as pd 
df = pd.DataFrame({"a":[1,3,1], "b":[10.0,8.0,6.0]})
df 
a b 0 1 10.0 1 3 8.0 2 1 6.0

筛选分组

# groupby 分组后,可以使用filter和agg对每个分组做聚合操作
# 筛选出满足条件(每分组的b列均值大于7.0且组内元素个数大于1)的分组
df.groupby(by=['a']).filter( lambda x: (x['b'].mean()>7.0) & (len(x['b']) > 1) )
a b 0 1 10.0 2 1 6.0

大家在看

京ICP备20031037号-1 | AI之家 | AI资讯 | Python200 | 数据分析