python - 如何在 pandas 中重置 cumsum streak

Question

我有以下数据框:

df = pd.DataFrame({'col1':['A','A','A','A','A','A','A','B','B','B','B','B','B','B'],'col2':[1,1,1,0,1,1,1,1,0,1,1,0,0,0],'col3':[.5,.5,.3,.2,.1,.4,.7,.4,.5,.3,.2,.4,.5,.6]})

我需要 col3 的运行总和，但只要 col1 发生变化或 col2 中有零，总和就应该重置。我应该返回如下内容: mock output

我尝试了以下方法:

df['cum'] = df.groupby([df.col1, df.col2]).col3.cumsum()

但是，它没有正确重置。它重置但继续对后续行求和。

Answer 1

在这里拍照

zeros = df.col2.eq(0)
vals = df.groupby(['col1', df.col2.ne(df.col2.shift()).cumsum()]).col3.cumsum()

df['id'] = np.where(zeros, 0, vals)

<小时/>

说明:

每当col2为0时，只需设置0。否则，只需设置分组值的cumsum。