python - 组内累计金额

Question

假设我有以下多索引数据框:

              A
H1   one      1
     two      0
     three    1
     four     2
H2   one      1
     two      4

我想在新列上计算组内的滚动累积总和，以便输出为:

              A   sum
H1   one      1     1
     two      0     1
     three    1     2
     four     2     4
H2   one      1     1   
     two      4     5

我怎样才能在 Pandas 中做到这一点？ (顺便说一句，我正在使用 master 分支)

Answer 1

我认为您需要做的就是将 level 参数与 groupby 一起使用(如本教程的 groupby with multiindex 部分所述):

>>> df["sum"] = df.groupby(level=0)["A"].cumsum()
>>> df
          A  sum

H1 one    1    1
   two    0    1
   three  1    2
   four   2    4
H2 one    1    1
   two    4    5