python - Pandas 中的累积总和从零开始，以除最后一个条目以外的所有组的总和结束

Question

在下面的数据框中，我想创建一个新列 C这将是 B 的累积和对于 A 中的每个组列，但这些总和必须从零开始，并且只添加值直到该组的倒数第二个条目。

  A B
0 1 5
1 1 6
2 2 3
3 2 4
4 2 5
5 3 2
5 3 7
6 4 3

所以，我的结果应该是:

  A B C
0 1 5 0
1 1 6 5
2 2 3 0
3 2 4 3
4 2 5 7
5 3 2 0
5 3 7 2
6 4 3 0

(我认为这个问题真的很明显，但不知何故我自己无法弄清楚，也无法在任何地方看到它。)

Answer 1

尝试:

df["C"] = df.groupby("A")["B"].transform(
    lambda x: x.shift().fillna(0).cumsum().astype(int)
)
print(df)

打印:

   A  B  C
0  1  5  0
1  1  6  5
2  2  3  0
3  2  4  3
4  2  5  7
5  3  2  0
5  3  7  2
6  4  3  0