集團內的累計金額

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

我怎么能在熊貓中做到這一點？ （我正在使用master分支BTW）

Answer 1

我認為，所有你需要做的是使用level與參數groupby （如所描述的與多指標GROUPBY教程的一部分）：

>>> 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