查找连续的重复项并列出它们在 python 中出现的位置的索引

Question

例如，我在 python 中有一个列表：mylist = [1,1,1,1,1,1,1,1,1,1,1,        0,0,1,1,1,1,0,0,0,0,0,        1,1,1,1,1,1,1,1,0,0,0,0,0,0]我的目标是找到一行中有五个或更多零的位置，然后列出发生这种情况的位置的索引，例如，输出将是：[17,21][30,35]这是我在此处提出的其他问题中尝试/看到的内容：def zero_runs(a):    # Create an array that is 1 where a is 0, and pad each end with an extra 0.    iszero = np.concatenate(([0], np.equal(a, 0).view(np.int8), [0]))    absdiff = np.abs(np.diff(iszero))    # Runs start and end where absdiff is 1.    ranges = np.where(absdiff == 1)[0].reshape(-1, 2)    return ranges    runs = zero_runs(list)这给出了输出：[0,10][11,12]...这基本上只是列出所有重复项的索引，我将如何将这些数据分成我需要的数据

Answer 1

您可以使用itertools.groupby，它将识别列表中的连续组：

from itertools import groupby

lst = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0]

groups = [(k, sum(1 for _ in g)) for k, g in groupby(lst)]

cursor = 0

result = []

for k, l in groups:

    if not k and l >= 5:

        result.append([cursor, cursor + l - 1])

    cursor += l

print(result)

输出

[[17, 21], [30, 35]]

Answer 2

使用itertools.groupbyand 的另一种方式enumerate。

首先找到零点和索引：

from operator import itemgetter

from itertools import groupby

zerosList = [

    list(map(itemgetter(0), g)) 

    for i, g in groupby(enumerate(mylist), key=itemgetter(1)) 

    if not i

]

print(zerosList)

#[[11, 12], [17, 18, 19, 20, 21], [30, 31, 32, 33, 34, 35]]

现在只需过滤zerosList：

runs = [[x[0], x[-1]] for x in zerosList if len(x) >= 5]

print(runs)

#[[17, 21], [30, 35]]