python - Pandas :求和列直到其他列中的条件满足

Question

我需要总结 value直到我打了一个 break .df = pd.DataFrame({'value': [1,2,3,4,5,6,7,8], 'break': [0,0,1,0,0,1,0,0]})

   value  break
0      1      0
1      2      0
2      3      1
3      4      0
4      5      0
5      6      1
6      7      0
7      8      0

预期输出

   value  break
0      6      1
1      15     1

我正在考虑一个小组，但我似乎无法使用它。我什至不需要 break列在最后。

Answer 1

你在正确的轨道上，在反向 cumsum 上尝试 groupby:

(df.groupby(df['break'][::-1].cumsum()[::-1],
            as_index=False, sort=False)
   .sum()
   .query('`break` != 0')  # remove this for full data 
)

输出:

   value  break
0      6      1
1     15      1