假设我有以下多索引数据框:
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)
答案 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