Поиск недостающих индексов данных с использованием python

Question

Каков оптимальный способ возврата индексов, где 1-й массив имеет отсутствующие данные. Отсутствующие данные представлены нулями. Данные могут быть действительно нулевыми, но не пропавшими без вести. Мы хотим вернуть индексы, где данные равны нулю для более чем или равно 3 местам за раз. Например, для массива [1,2,3,4,0,1,2,3,0,0,0,1,2,3] функция должна возвращать только индексы для второго сегмента, где есть нули, а не первые пример.

Это на самом деле вопрос интервью :) Задача состоит в том, чтобы сделать наиболее effeciently в одной строке

Answer 1

Отслеживайте подсчета нулей в текущем периоде. Затем, если заканчивается пробег, который имеет по крайней мере три нуля, вычисляет индексы.

def find_dx_of_missing(a): 
    runsize = 3 # 3 or more, change to 4 if your need "more than 3" 
    zcount = 0 
    for i, n in enumerate(a): 
     if n == 0: 
      zcount += 1 
     else: 
      if zcount >= runsize: 
       for j in range(i - zcount, i): 
        yield j 
      zcount = 0 
    if zcount >= runsize: # needed if sequence ends with missing 
     i += 1 
     for j in range(i - zcount, i): 
      yield j 

Примеры:

>>> a = [1,2,3,4,0,1,2,3,0,0,0,1,2,3] 
>>> list(find_dx_of_missing(a)) 
[8, 9, 10] 

>>> a = [0,0,0,3,0,5,0,0,0,0,10,0,0,0,0,0] 
>>> list(find_dx_of_missing(a)) 
[0, 1, 2, 6, 7, 8, 9, 11, 12, 13, 14, 15] 

Edit: Поскольку вам нужен один лайнер здесь два кандидата, предполагающие a в ваш список и n является наименьшим пробег нулей рассчитывать недостающие данные:

[v for vals in (list(vals) for iszeros, vals in itertools.groupby(xrange(len(a)), lambda dx, a=a: a[dx]==0) if iszeros) for v in vals if len(vals) >= n] 

Или

sorted({dx for i in xrange(len(a)-n+1) for dx in xrange(i, i+n) if set(a[i:i+n]) == {0}})