python - 一旦满足条件，累加 1 并重置为 0

Question

目前我在下面有一个数据集，如果 ColA 为 0，我尝试累加该值，而如果 ColA 再次为 1，则将值重置为 0(再次重新开始计数)。

ColA  
 1        
 0        
 1        
 1        
 0        
 1        
 0         
 0        
 0        
 1        
 0        
 0        
 0        

我的预期结果如下。

ColA  Accumulate
 1        0
 0        1
 1        0 
 1        0 
 0        1
 1        0
 0        1 
 0        2 
 0        3
 1        0
 0        1
 0        2
 0        3

我目前使用的代码

test['Value'] = np.where ( (test['ColA']==1),test['ColA'].cumsum() ,0)


ColA   Value
 1        0
 0        1
 1        0 
 1        0 
 0        2
 1        0
 0        3 
 0        4 
 0        5
 1        0
 0        6
 0        7
 0        8

Answer 1

使用cumsum如果性能很重要:

a = df['ColA'] == 0
cumsumed = a.cumsum()
df['Accumulate'] = cumsumed-cumsumed.where(~a).ffill().fillna(0).astype(int)

print (df)
    ColA  Accumulate
0      1           0
1      0           1
2      1           0
3      1           0
4      0           1
5      1           0
6      0           1
7      0           2
8      0           3
9      1           0
10     0           1
11     0           2
12     0           3