我有一个python问题,我似乎无法解决。以下数据框说明了在给定年份和月份内花钱的人:
ID | |||||||||||
---|---|---|---|---|---|---|---|---|---|---|---|
134 | 134 | 134 | 135 | 135 | 135 |
您可以使用cumsum() + 100
:
df["NewColumn"] = df.groupby(["ID"])["Amount"].cumsum()+100
输出:
ID Year Month Amount NewColumn
0 134 2020 11 -199 -99
1 134 2020 12 -50 -149
2 134 2021 1 40 -109
3 135 2020 11 -365 -265
4 135 2020 12 -23 -288
5 135 2021 1 400 112
PS:您的预期输出有一个小差异,因为您从-200
开始,但在数据中是-199
:-(