Python：具有更改键的累积和

Question

我有一个数据表，例如：

F(1)  F(2)  F(3)  Amount
A     B     C     100
A     B     C     100
A     B     C     100
D     E     F     300
D     E     F     150
G     H     I     100
G     H     I     200

我想产生一个新列，显示字段“金额”的累积总和，但是只要列F（1），F（2）和F（3）的键更改，该列就会重置为0。

即我想创建以下输出（无虚线！）

F(1)  F(2)  F(3)  Amount  CumSum
A     B     C     100     100
A     B     C     100     200
A     B     C     100     300
------------------------------ resets to zero as key changes
D     E     F     300     300
D     E     F     150     450
------------------------------ resets to zero as key changes
G     H     I     100     100
G     H     I     200     300

该表中可能有多达一百万行，因此我正在寻找可靠的实现。 熊猫在这里是前进的方向吗？ 我以前没有用过熊猫，但很高兴探索。

Answer 1

按您的关键列分组并致电cumsum：

df['CumSum'] = df.groupby(['F(1)', 'F(2)', 'F(3)'])['Amount'].cumsum()

df
Out: 
  F(1) F(2) F(3)  Amount  CumSum
0    A    B    C     100     100
1    A    B    C     100     200
2    A    B    C     100     300
3    D    E    F     300     300
4    D    E    F     150     450
5    G    H    I     100     100
6    G    H    I     200     300