Cumsum是现有Pandas数据中的新列

时间:2017-01-25 18:48:10

标签: python pandas dataframe cumsum

我将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被累积和替换。但是,我的要求是将累积金额作为新列添加到现有数据框中。

谢谢

2 个答案:

答案 0 :(得分:10)

只需在cumsum pandas.Series上应用df['SUM_C']并将其分配到新列:

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

答案 1 :(得分:0)

概述:您可以使用数据框聚合并将其传递给用户定义的函数

["d", "d", "e", "d", "e"]