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