Pandas：数据框 cumsum ，如果其他列为假则重置 [重复]

Question

我有一个包含 2 列的数据框，这里的目标很简单；如果行列设置为 False，则重置 df.cumsum()；

df

      value      condition
0       1            1
1       2            1
2       3            1
3       4            0
4       5            1

想要的结果如下：

df

      value      condition
0       1            1
1       3            1
2       6            1
3       4            0
4       9            1

如果我按照本文Python pandas cumsum() reset after hitting max 中的描述循环数据帧，我可以达到想要的结果，但我正在寻找一种使用 pandas 标准函数的更加矢量化的方式

Answer 1

怎么样：

df['cSum'] = df.groupby((df.condition == 0).cumsum()).value.cumsum()

输出：

   value  condition  cSum
0      1          1     1
1      2          1     3
2      3          1     6
3      4          0     4
4      5          1     9

您会将连续的行组合在一起，直到在condition 列中遇到0，然后在每个组中分别应用cumsum。