def rcount(a):
    without_reset = (a == 1).cumsum()
reset_at = a == 0
overcount = np.maximum.accumulate(without_reset * reset_at)
result = without_reset - overcount
return result
Совокупное количество в NumPy без итерации