python - 没有当前行的Python中的累积和

Question

我需要在不计算当前行的情况下找到前 3 行的累积和，这里有一个简短的示例：

ID	和
一个	10
乙	5
乙	4
乙	1
乙	2
C	1
C	3

这就是我想要达到的结果：

ID	和
一个	0
乙	0
乙	5
乙	9
乙	10
C	0
C	3

标签： pythonpandasdataframerolling-computation

Answer 1

你可以试试：

df['SUM'] = (df.groupby('ID')['SUM']
               .transform(lambda x: x.rolling(4,min_periods=1).sum())
               .sub(df['SUM'])
            )

输出：

  ID  SUM
0  A    0
1  B    0
2  B    5
3  B    9
4  B   10
5  C    0
6  C    1