Подробнее pythonic способ фильтрации стабильных значений в списке

Question

Я написал функцию, которая позволяет мне просматривать список, сравнивать значения с предшественниками и утверждать, в какой момент список становится «стабильным» для определенного количества записей. Значения в списке представляют собой сигнал, который может достигать или не достигать стабильной точки. я пришел с этим:

def unstableFor(points, maxStable): 
    count = 0; 
    prev = points[0] 
    for i in range(1, len(points)): 
     if points[i] == prev: 
      count = count + 1 
     else: 
      count = 0 
      prev = points[i] 
     if count >= maxStable: 
      return i 
    return len(points) - 1 

Возвращаемое значение затем используется вызывающим убирания последнюю часть списка.

Выполняет свою работу, однако, я очень недоволен тем, насколько громоздким оно выглядит. Можете ли вы представить себе более питоновский, возможно, чисто функциональный способ выполнения этой операции фильтрации?

Answer 1

Использование перечисления и архивирования:

def unstableFor (points, threshold): 
    for i, (a, b) in enumerate(zip(points, points[1:])): 
     count = count + 1 if a == b else 0 
     if count >= threshold: 
      return i 
    return i 

Answer 2

Вот эскиз функционального подхода. Это немного загадочно. Действительно, я, скорее всего, буду использовать ваш подход (используя enumerate, как идиоматический способ, а не range(len(x))). Во всяком случае, предположим, max_stable 3:

>>> from itertools import groupby 
>>> grouped = groupby(enumerate(x), lambda i_e: i_e[1]) 
>>> gen = (g for g in map(lambda e: list(e[1]), grouped) if len(g) >= 3) 
>>> run = next(gen) 
>>> run[2][0] 
10 

Здесь очищается:

>>> from operator import itemgetter 
>>> from itertools import islice 
>>> def unstable_for(points, max_stable): 
...  grouped = groupby(enumerate(points), itemgetter(1)) 
...  gen = (g for g in (tuple(gg) for _, gg in grouped) if len(g) >= max_stable) 
...  run = tuple(islice(gen,1)) 
...  if len(run) == 0: 
...   return len(points) - 1 
...  else: 
...   return run[0][max_stable - 1][0] 
... 
>>> x 
[1, 2, 3, 3, 4, 5, 6, 7, 8, 8, 8, 8, 8, 9] 
>>> unstable_for(x, 3) 
10 
>>> unstable_for(x, 2) 
3 
>>> unstable_for(x, 1) 
0 
>>> unstable_for(x, 20) 
13 
>>> 

Не очень элегантно. Опять же, я бы пошел с императивным решением. Возможно, у кого-то есть более элегантное функциональное решение.

Answer 3

Ваш код выглядит хорошо для меня: его легко читать и понимать. Я бы просто удалить некоторые повторы, чтобы сделать его выглядеть следующим образом:

def unstableFor(points, maxStable): 
    prev = None # assuming None is not member of points 
    for i, point in enumerate(points): 
     if point == prev: 
      count = count + 1 
     else: 
      count = 0 
      prev = point 
     if count >= maxStable: 
      break 
    return i