汇总作为现有Pandas数据中的新列

Question

我有一个定义为的熊猫数据框：

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

在group（）上的熊猫中使用cumsum显示了生成新数据帧的可能性，其中列名SUM_C被替换为累积和。但是，我的要求是将累加总和作为新列添加到现有数据框中。

谢谢

Answer 1

刚申请cumsum的pandas.Series df['SUM_C']，并将其分配给一个新的列：

df['CUMSUM_C'] = df['SUM_C'].cumsum()

结果：

df
Out[34]: 
   A  B  SUM_C  CUMSUM_C
0  1  1     10       10
1  1  2     20       30