Python在循环时计算运行总计

Question

我正在尝试将计算的运行总计列添加到基于另一列的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  

Answer 1

你需要：

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