python - 对 pandas 系列中的多索引级别求和

Question

使用 python 中的 Pandas 包，我想对具有 3 级多索引的系列中的一个级别求和(边缘化)以生成具有 2 级多索引的系列。例如，如果我有以下内容:

ind = [tuple(x) for x in ['ABC', 'ABc', 'AbC', 'Abc', 'aBC', 'aBc', 'abC', 'abc']]
mi = pd.MultiIndex.from_tuples(ind)
data = pd.Series([264, 13, 29, 8, 152, 7, 15, 1], index=mi)

A  B  C    264
      c     13
   b  C     29
      c      8
a  B  C    152
      c      7
   b  C     15
      c      1

我想对变量 C 求和以产生以下输出:

A  B    277
   b     37
a  B    159
   b     16

Pandas 中最好的方法是什么？

Answer 1

如果您知道您总是希望在前两个级别上进行聚合，那么这很容易:

In [27]: data.groupby(level=[0, 1]).sum()
Out[27]:
A  B    277
   b     37
a  B    159
   b     16
dtype: int64