python - Pandas groupby 对象过滤

Question

我有一个 Pandas 数据框

df.columns
Index([u’car_id’,u’color’,u’make’,u’year’)]

我想创建一个新的 FILTERABLE 对象，其中包含每个组(颜色、品牌、年份)的计数；

grp = df[[‘color’,’make’,’year’]].groupby([‘color’,’make’,’year’]).size()

它会返回这样的东西

color   make   year     count
black   honda  2011   416

我希望能够过滤它，但是当我尝试这样做时:

grp.filter(lambda x: x[‘color’]==‘black’)

我收到这个错误

TypeError: 'function' object is not iterable

我如何利用“groupby”对象来过滤掉行？

Answer 1

我认为您需要添加 reset_index然后输出是DataFrame。最后使用boolean indexing :

df = df[['color','make','year']].groupby(['color','make','year'])
                                .size()
                                .reset_index(name='count')


df1 = df[df.color == 'black']