我在python中有一个类似以下内容的df:
'A'
0
1
0
0
1
1
1
1
0
我想创建另一列,将A列中的累积1相加,如果A列中的值再次变为0,则重新开始。因此,期望输出:
'A' 'B'
0 0
1 1
0 0
0 0
1 1
1 2
1 3
1 4
0 0
这是我正在尝试的,但它只是复制A列:
df.B[df.A ==0] = 0
df.B[df.A !=0] = df.A + df.B.shift(1)
让我们用{}{}来做{}
将^{} 与^{} 和^{} 一起使用:
相关问题 更多 >
编程相关推荐