Question

这可能很容易，但我希望计算python中列表中连续正值出现的长度。例如，我有一个，我希望返回b：

a=[0,0,1,1,1,1,0,0,1,0,1,1,1,0]

b=[0,0,4,4,4,4,0,0,1,0,3,3,3,0]

我在Counting consecutive positive value in Python array上注意到类似的问题，但这只会返回连续的计数，但不会返回所属组的长度。

由于

Answer 1

这类似于run length encoding问题，所以我从Rosetta代码页中借用了一些想法：

import itertools
a=[0,0,1,1,1,1,0,0,1,0,1,1,1,0]

b = []
for item, group in itertools.groupby(a):
    size = len(list(group))
    for i in range(size):
        if item == 0:
            b.append(0)
        else:
            b.append(size)

b
Out[8]: [0, 0, 4, 4, 4, 4, 0, 0, 1, 0, 3, 3, 3, 0]

Answer 2

经过这么多尝试后，最后提出了这两行。

In [9]: from itertools import groupby

In [10]: lst=[list(g) for k,g in groupby(a)]

In [21]: [x*len(_lst) if x>=0 else x for _lst in lst for x in _lst]
Out[21]: [0, 0, 4, 4, 4, 4, 0, 0, 1, 0, 3, 3, 3, 0]

Answer 3

这是一种方法。

基本前提是，当连续运行正值时，它将记住这些正值的所有指数。一旦它达到零，它将回溯并用运行的长度替换所有正值。

a=[0,0,1,1,1,1,0,0,1,0,1,1,1,0]

glob = []
last = None
for idx, i in enumerate(a):
    if i>0:
        glob.append(idx)
    if i==0 and last != i:
        for j in glob:
            a[j] = len(glob)
        glob = []

# > [0, 0, 4, 4, 4, 4, 0, 0, 1, 0, 3, 3, 3, 0]

python连续计数出现的长度

3 个答案: