python - 我需要获取某个时间窗口和某个步骤的金额

Question

我需要获取某个时间窗口和某个步骤的金额。下面是我的数据集。

    TransactionDate PartyId AmountEur
0   2018-04-28  15754387    8.46
1   2018-04-29  15754387    31.82
2   2018-04-30  15754387    1389.43
3   2018-05-01  15754387    134.12
4   2018-05-02  15754387    44.16
5   2018-05-03  15754387    65.62
6   2018-05-04  15754387    20.50
7   2018-05-05  15754387    23.92
8   2018-05-07  15754387    17.41
9   2018-05-08  15754387    10.32

例如对于 window=4 和 step=1，应该得到：

    TransactionDate PartyId AmountEur Res
0   2018-04-28  15754387    8.46      8.46 (8.64)
1   2018-04-29  15754387    31.82     40.28 (8.46+31.82)
2   2018-04-30  15754387    1389.43   1429.71 (8.46+31.82+1389.43)
3   2018-05-01  15754387    134.12    1563.83 (8.46+31.82+1389.43+134.12)
4   2018-05-02  15754387    44.16     1599.53 (31.82+1389.43+134.12+44.16)
5   2018-05-03  15754387    65.62     1633.33 (1389.43+134.12+44.16+65.62)
6   2018-05-04  15754387    20.50     264.4  (134.12+44.16+65.62+20.50)
7   2018-05-05  15754387    23.92     154.2 (44.16+65.62+20.50+23.92)
8   2018-05-07  15754387    17.41     127.45 (65.62+20.50+23.92+17.41)
9   2018-05-08  15754387    10.32     72.15 (20.50+23.92+17.41+10.32)

感谢所有试图帮助我的人！

标签： pythonpandas

Answer 1

你可以用滚动总和来做到这一点：

df['Res'] = df.AmountEur.rolling(4, 1).sum()

在这里查看更多信息。