a = [0,0,1,1,1,0,0,1,0,1,1] b = [0,0,0,0,0,0,0,0,0,0,0] for i in range(len(a)): if a[i] == 1: b[i] = b[i-1] + 1 else: b[i] = 0