Перебирать список, начиная с определенного элемента

Question

Скажите, у меня есть список:

l = [1, 2, 3, 4]

И я хочу пройти через это. Обычно, это будет что-то вроде этого,

1, 2, 3, 4, 1, 2, 3, 4, 1, 2...

Я хочу иметь возможность начать с определенной точки цикла, не обязательно индекса, но, возможно, соответствия элемента. Скажем, я хотел бы начать с любого элемента в списке ==4, тогда вывод будет,

4, 1, 2, 3, 4, 1, 2, 3, 4, 1...

Как мне это сделать?

Answer 1

Посмотрите на модуль itertools .Предоставляет все необходимые функции.

from itertools import cycle, islice, dropwhile

L = [1, 2, 3, 4]

cycled = cycle(L)  # cycle thorugh the list 'L'
skipped = dropwhile(lambda x: x != 4, cycled)  # drop the values until x==4
sliced = islice(skipped, None, 10)  # take the first 10 values

result = list(sliced)  # create a list from iterator
print(result)

Вывод:

[4, 1, 2, 3, 4, 1, 2, 3, 4, 1]

Answer 2

Используйте арифметический оператор mod. Предположим, вы начинаете с позиции k, тогда k следует обновить следующим образом:

k = (k + 1) % len(l)

Если вы хотите начать с определенного элемента, а не с индекса, вы всегда можете посмотреть его как k = l.index(x), где x - желаемый элемент.

Answer 3

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

def cycle(my_list, start_at=None):
    start_at = 0 if start_at is None else my_list.index(start_at)
    while True:
        yield my_list[start_at]
        start_at = (start_at + 1) % len(my_list)

Это вернет (бесконечный) итератор, зацикливающий ваш список. Чтобы получить следующий элемент в цикле, вы должны использовать оператор next:

>>> it1 = cycle([101,102,103,104])
>>> next(it1), next(it1), next(it1), next(it1), next(it1)
(101, 102, 103, 104, 101) # and so on ...
>>> it1 = cycle([101,102,103,104], start_at=103)
>>> next(it1), next(it1), next(it1), next(it1), next(it1)
(103, 104, 101, 102, 103) # and so on ...

Answer 4

import itertools as it
l = [1, 2, 3, 4]
list(it.islice(it.dropwhile(lambda x: x != 4, it.cycle(l)),  10))
# returns: [4, 1, 2, 3, 4, 1, 2, 3, 4, 1]

поэтому итератор, который вам нужен:

it.dropwhile(lambda x: x != 4, it.cycle(l))

Answer 5

Хм, http://docs.python.org/library/itertools.html#itertools.cycle не имеет такого начального элемента.

Возможно, вы все равно просто начнете цикл и отбросите первые элементы, которые вам не нравятся.

Answer 6

Еще одна странная опция заключается в том, что циклический просмотр списков может быть выполнен в обратном направлении .Например:

# Run this once
myList = ['foo', 'bar', 'baz', 'boom']
myItem = 'baz'

# Run this repeatedly to cycle through the list
if myItem in myList:
    myItem = myList[myList.index(myItem)-1]
    print myItem