计算列表中连续相同值的数量

Question

我有一个二进制值列表：1 =可同步，0 =不同步。 每个二进制值表示一个人当天是否可以同步。

person1=[1,1,0,1,1,0,1,1,1]

我想列出另一个清单，计算一个人可以连续同步多少天。 当出现0时，计数器基本上会复位。

因此，对于上面的示例，输出如下所示：

person1=[1,2,0,1,2,0,1,2,3]

我的问题是，只要列表遇到0，我都不知道如何重置它。 我尝试了几种方法，但都无济于事。

x=[1,1,1,1,1]
y=[]

for index, value in enumerate(x):
     y.append(value+sum(x[:index]))

print(y)
...

[1, 2, 3, 4, 5]

任何帮助表示赞赏。 我认为使用递归可能会有所帮助。

Answer 1

做这个：

i = 0
y = []
for value in person1:
    if value:
        i += 1
    else:
        i = 0
    y.append(i)

没有理由创建一个列表。 您只需要跟踪柜台。

Answer 2

使用itertools.groupby ，您可以生成一个包含同步总和的列表。

import itertools

person1=[1,1,0,1,1,0,1,1,1]
syncs = [sum(v) for _, v in itertools.groupby(person1)]

输出：

[2, 0, 2, 0, 3]

而且，如果您只是想让人们连续几天都可以“同步”，请执行以下操作：

max_syncs = max[sum(v) for _, v in itertools.groupby(person1) if _]