Перезаписывание вложенных операторов if в более Pythonic

Question

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

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

def sequence_in_list(seq, lst): 
    m, n = len(lst), len(seq) 
    for i in xrange(m): 
     for j in xrange(n): 
      if lst[i] == seq[j]: 
       if lst[i+1] == seq[j+1]: 
        if lst[i+2] == seq[j+2]: 
         return lst[i+3] 

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

Answer 1

С вами сравнивают последовательные индексы, и предполагая lst и seq одного и того же типа, вы можете использовать нарезку:

def sequence_in_list(seq, lst): 
    m, n = len(lst), len(seq) 
    for i in xrange(m): 
     for j in xrange(n): 
      if lst[i:i+3] == seq[j:j+3]: 
       return lst[i+3] 

Если последовательности а re разного типа, вы должны преобразовать их в общий тип перед выполнением сравнения (например, lst[i:i+3] == list(seq[j:j+3]) будет работать, если seq - это строка и lst - это список).

В качестве альтернативы, если последовательности не поддерживают нарезку, вы можете использовать встроенный в all для проверки нескольких условий:

def sequence_in_list(seq, lst): 
    m, n = len(lst), len(seq) 
    for i in xrange(m): 
     for j in xrange(n): 
      if all(lst[i+k] == seq[j+k] for k in range(3)): 
       return lst[i+3] 

Если вы хотите продлить проверку более 10 показателей, вместо 3, просто изменить range(3) до range(10).

Side Примечание: исходный код будет поднимать IndexError в какой-то момент, так как доступ list[i+1] где i может быть len(list) - 1. Вышеприведенный код не вызывает никаких ошибок, так как нарезка может дать срез короче, чем разница индексов, meainig, что seq[j:j+3] может иметь менее 3 элементов. Если это проблема, вы должны настроить индексы, по которым вы выполняете итерацию.

Последнее замечание: не используйте имя list, поскольку оно затеняет встроенное имя.

Answer 2

Я хотел бы сделать это с помощью генератора и нарезка:

sequence = [1, 2, 3, 5, 1, 2, 3, 6, 1, 2, 3] 
pattern = [1, 2, 3] 

def find_item_after_pattern(sequence, pattern): 
    n = len(pattern) 

    for index in range(0, len(sequence) - n): 
     if pattern == sequence[index:index + n]: 
      yield sequence[index + n] 

for item in find_item_after_pattern(sequence, pattern): 
    print(item) 

И вы получите:

5 
6 

Функция не слишком эффективна и не будет работать для бесконечных последовательностей, но он короткий и общий.

Answer 3

Вы можете совместить список понимание с нарезкой, чтобы сделать сравнение более читаемым:

n, m = len(lst), len(seq) 
[lst[j+3] for i in range(m-2) for j in range(n-2) if seq[i:i+3] == lst[j:j+3]] 

Конечно, есть более эффективные способы сделать это, но это просто, коротко и питон стиле.