我有一个pandas数据框定义为:
A B SUM_C
1 1 10
1 2 20
我想做一个SUM_C的累积和,并将其作为新列添加到同一个数据帧.换句话说,我的最终目标是拥有如下所示的数据框:
A B SUM_C CUMSUM_C
1 1 10 10
1 2 20 30
Using cumsum in pandas on group()显示了生成新数据帧的可能性,其中列名称SUM_C被累积和替换.但是,我的问题是将累积和作为新列添加到现有数据框中.
谢谢
最佳答案 只需在pandas.Series df [‘SUM_C’]上应用cumsum并将其分配给新列:
df['CUMSUM_C'] = df['SUM_C'].cumsum(axis = 0)
结果:
df
Out[34]:
A B SUM_C CUMSUM_C
0 1 1 10 10
1 1 2 20 30