python – Cumsum作为现有Pandas数据中的新列

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