ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

python – Pandas:如何分组并对MultiIndex求和

2019-08-24 15:55:12  阅读:255  来源: 互联网

标签:multi-index python pandas


我有一个带有分类属性的数据框,其中索引包含重复项.我试图找到索引和属性的每个可能组合的总和.

x = pd.DataFrame({'x':[1,1,3,3],'y':[3,3,5,5]},index=[11,11,12,12])
y = x.stack()
print(y)
print(y.groupby(level=[0,1]).sum())

产量

11  x    1
    y    3
    x    1
    y    3
12  x    3
    y    5
    x    3
    y    5
dtype: int64
11  x    1
    y    3
    x    1
    y    3
12  x    3
    y    5
    x    3
    y    5
dtype: int64

堆栈和组合总和是一样的.

但是,我期待的是

11  x    2
11  y    6
12  x    6
12  y    10

编辑2:

x = pd.DataFrame({'x':[1,1,3,3],'y':[3,3,5,5]},index=[11,11,12,12])
y = x.stack().groupby(level=[0,1]).sum()
print(y.groupby(level=[0,1]).sum())

输出:

11  x    1
    y    3
    x    1
    y    3
12  x    3
    y    5
    x    3
    y    5
dtype: int64

EDIT3:
已记录一个问题
https://github.com/pydata/pandas/issues/10417

解决方法:

使用pandas 0.16.2和Python 3,我能够通过以下方式获得正确的结果:

x.stack().reset_index().groupby(['level_0','level_1']).sum()

哪个产生:

                    0
level_0 level_1 
     11       x     2
              y     6
     12       x     6
              y     10

然后,您可以使用reindex()和columns将索引和列名称更改为更合适的名称.

根据我的研究,我同意原始方法的失败似乎是一个错误.我认为这个bug出现在Series上,这就是x.stack()产生的.我的解决方法是通过reset_index()将Series转换为DataFrame.在这种情况下,DataFrame不再具有MultiIndex – 我只是对标记列进行分组.

要确保对具有MultiIndex的DataFrame进行分组和求和,您可以尝试使用它来获得相同的正确输出:

x.stack().reset_index().set_index(['level_0','level_1'],drop=True).\
groupby(level=[0,1]).sum()

在解决错误之前,这些变通办法中的任何一个都应该处理好事情.

我想知道这个bug是否与在Series与DataFrame上创建的MultiIndex实例有关.例如:

In[1]: obj = x.stack()
       type(obj)
Out[1]: pandas.core.series.Series

In[2]: obj.index
Out[2]: MultiIndex(levels=[[11, 11, 12, 12], ['x', 'y']],
           labels=[[0, 0, 1, 1, 2, 2, 3, 3], [0, 1, 0, 1, 0, 1, 0, 1]])

In[3]: obj = x.stack().reset_index().set_index(['level_0','level_1'],drop=True)
       type(obj)
Out[3]: pandas.core.frame.DataFrame

In[4]: obj.index
Out[4]: MultiIndex(levels=[[11, 12], ['x', 'y']],
           labels=[[0, 0, 0, 0, 1, 1, 1, 1], [0, 1, 0, 1, 0, 1, 0, 1]],
           names=['level_0', 'level_1'])

请注意DataFrame上的MultiIndex如何更准确地描述级别.

标签:multi-index,python,pandas
来源: https://codeday.me/bug/20190824/1709607.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有