Python在循环时计算运行总计

时间:2018-06-13 13:57:26

标签: python pandas dataframe

我正在尝试将计算的运行总计列添加到基于另一列的Pandas数据框中。在下面的示例中,计算列(Running Total)应该保留员工薪水的运行总计,然后在完成员工A的循环时以新的运行总计重新启动。

输入:

Employee    |    Pay   
-----------------------  
A      |       1        
A      |       2         
A      |       3       
B      |       2   
B      |       3    

输出:

Employee    |    Pay   |  Running Total
----------------------------------------  
A      |       1       |      1  
A      |       2       |      3  
A      |       3       |      6  
B      |       2       |      2  
B      |       3       |      5  

1 个答案:

答案 0 :(得分:0)

你需要:

df['Running Total'] = df.groupby('Employee').cumsum()

输出:

  Employee  Pay  Running Total
0  A          1              1
1  A          2              3
2  A          3              6
3  B          2              2
4  B          3              5