我有一个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