我有一份 list ,上面有以下编号:
[0, 0, 0, 0, 1, 1, 0, 0, 1]
我想做的是计算我有多少个连续的1,并生成一个新列表,如下所示:
[0, 0, 0, 0, 1, 2, 0, 0, 1]
有没有比这种方法更紧凑的方法?
lst = [0, 0, 0, 0, 1, 1, 0, 0, 1]
new_lst = []
counter = 0
for i in range(len(lst)):
if (lst[i] == 1): counter+=1
else:counter = 0
new_lst.append(counter)
print(new_lst)