Python，Pandas Dataframe 在分组后取回索引

Posted 2023-03-11

【中文标题】Python，Pandas Dataframe 在分组后取回索引

【英文标题】：Python, Pandas Dataframe get the index back after a group by

【发布时间】：2015-01-26 19:57:47

【问题描述】：

我有一个名为“class_price_df”的 Pandas 数据框：

                           email                   cat           class
0               [e-mail]     Mobiles & Tablets      1
1                    [e-mail]     Mobiles & Tablets      4
2                   [e-mail]     Mobiles & Tablets      2
3                   [e-mail]     Mobiles & Tablets      4

我按“电子邮件”和“猫”分组以获得最大“类”：

class_price_df = class_price_df.groupby(['email','cat']).max().unstack('cat').fillna(0)

但是输出是：

cat                               Computers & Laptops  Consumer Electronics   
email                                                                         
[e-mail]                               2                     0   
[e-mail]                             0                     2   
[e-mail]                        0                     0   
[e-mail]                                      0                     0   
[e-mail]                                 0                     4   
[e-mail]                                    1                     0   
[e-mail]                              4                     0   
[e-mail]                                 0                     0   

我怎样才能找回我的“索引”并获得类似于以下内容的输出：

                email                 Computers & Laptops        Consumer Electronics
0      [e-mail]                   2                             0  
1      [e-mail]                 0                             2 
2      [e-mail]            0                             0   
3      [e-mail]                          0                             4

【参考方案1】：

只需使用reset_index 方法：

class_price_df.reset_index(inplace=True)

【讨论】：

谢谢，但是当我在做 class_price_df.columns 时，我有 MultiIndex(levels=[[u'class', u'email'], [u'Automotive & Gadgets', u'Cameras' , u'Computers & Laptops'.......) 我怎样才能得到 Index([u'email', u'Automotive & Gadgets', u'Cameras', u'Computers & Laptops'..... ....) ?

@user1754181 您可以在调用 reset_index 时使用 int 或字符串名称指定级别：pandas.pydata.org/pandas-docs/stable/generated/… 例如 class_price_df.reset_index(inplace=True, level='class')

感谢@EdChum，但我仍然有 MultiIndex(levels=[[u'class', u'email'], [u'Automotive & Gadgets', u'Cameras', u'Computers & Laptops '