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