Pandas：如何对 MultiIndex 进行分组和求和

Question

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

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

stack 和 group by sum 是一样的。

然而，我期望的是

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

Answer 1

使用 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。

根据我的研究，我同意原始方法的失败似乎是一个错误。我认为错误在Series，这就是x.stack()产生的原因。我的解决方法是将 theSeries变成DataFramevia reset_index()。在这种情况下，DataFrame不再有MultiIndex- 我只是在标记的列上分组。

为了确保分组并在总结工作DataFrame用MultiIndex，你可以试试这个得到同样正确的输出：

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

这些解决方法中的任何一个都应该处理问题，直到错误得到解决。

我想知道该错误是否与MultiIndex在 aSeries与 a上创建的实例有关DataFrame。例如：

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'])

请注意如何MultiIndex在DataFrame更准确地描述了水平。

Answer 2

sum 允许您指定要在 MultiIndex 数据框中求和的级别。

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

y.sum(level=[0,1])

11  x     2
    y     6
12  x     6
    y    10