集团内的累计金额

时间:2014-05-22 23:02:11

标签: python pandas

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

              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

我怎么能在熊猫中做到这一点? (我正在使用master分支BTW)

1 个答案:

答案 0 :(得分:3)

我认为您需要做的就是将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