Итератор кругового списка в Python

101

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

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

Есть ли удобный способ сделать это в Python?

python list iterator user443854
источник

Ответы:

161

Используйте itertools.cycle, вот его точное предназначение:

from itertools import cycle

lst = ['a', 'b', 'c']

pool = cycle(lst)

for item in pool:
    print item,

Вывод:

a b c a b c ...

(Петли вечны, очевидно)

Чтобы вручную продвигать итератор и извлекать из него значения одно за другим, просто вызовите next(pool):

>>> next(pool)
'a'
>>> next(pool)
'b'
Лукас Граф
источник
2
Вы печатаете элементы в цикле. Что хочу выйти из цикла и вернуться позже? (Я хочу начать с того места, где остановился).
user443854 01
7
@ user443854 используйте pool.next()для получения единственного следующего элемента из цикла
Джейкоб Кролл
4
@ user443854 FWIW, это гораздо лучший ответ, чем мой. Нет причин возвращаться к повторной реализации библиотечных функций!
Джейкоб Кролл
5
pool.next () у меня не работал, только next (pool). Наверное, из-за Python 3?
fjsj
6
@fjsj, что правильно, на Python 3 вам нужно использовать next(iterator)(который, кстати, также отлично работает на Python 2.x, и, следовательно, является канонической формой, которую следует использовать). См. Виден ли генератор.next () в python 3.0? для более подробного объяснения. Соответственно обновил свой ответ.
Лукас Граф
55

Правильный ответ - использовать itertools.cycle . Но предположим, что такой библиотечной функции не существует. Как бы вы это реализовали?

Используйте генератор :

def circular():
    while True:
        for connection in ['a', 'b', 'c']:
            yield connection

Затем вы можете использовать forоператор для бесконечной итерации или вызвать next()для получения единственного следующего значения из итератора генератора:

connections = circular()
next(connections) # 'a'
next(connections) # 'b'
next(connections) # 'c'
next(connections) # 'a'
next(connections) # 'b'
next(connections) # 'c'
next(connections) # 'a'
#....
Джейкоб Кролл
источник
Ницца! Как он знает, что нужно начинать заново, когда список исчерпан?
user443854 01
1
@ user443854 while Trueсредство повторять вечно
Джейкоб Кролл
3
@juanchopanza: Ага; itertools.cycleэто лучший ответ. Это показывает, как можно было бы написать ту же функцию, если itertoolsона недоступна :)
Джейкоб Кролл
Сохраняет ли простой генератор также копию каждого элемента, как это itertools.cycleделает? Или простой генератор будет более эффективным с точки зрения памяти? Согласно cycleдокументам :Note, this member of the toolkit may require significant auxiliary storage (depending on the length of the iterable).
dthor
2
@dthor этот генератор создает список из трех элементов и обрабатывает его, затем уничтожает список и создает новый навсегда. Эта документация cycleподразумевает, что итерация ввода преобразуется в listдо запуска генератора, поскольку iterableона « годна только для одного прохода по набору значений».
Джейкоб Кролл
9

Или можно сделать так:

conn = ['a', 'b', 'c', 'd', 'e', 'f']
conn_len = len(conn)
index = 0
while True:
    print(conn[index])
    index = (index + 1) % conn_len

печатает abcdefab c ... навсегда

viky.pat
источник
4

вы можете сделать это с помощью append(pop())цикла:

l = ['a','b','c','d']
while 1:
    print l[0]
    l.append(l.pop(0))

или for i in range()цикл:

l = ['a','b','c','d']
ll = len(l)
while 1:
    for i in range(ll):
       print l[i]

или просто:

l = ['a','b','c','d']

while 1:
    for i in l:
       print i

все из которых печатают:

>>>
a
b
c
d
a
b
c
d
...etc.

из трех я был бы склонен к подходу append (pop ()) как функции

servers = ['a','b','c','d']

def rotate_servers(servers):
    servers.append(servers.pop(0))
    return servers

while 1:
    servers = rotate_servers(servers)
    print servers[0]
легкое присутствие
источник
1
Проголосовать за это, потому что это помогло мне с совершенно другим вариантом использования, когда я просто хочу перебирать список несколько раз, каждый раз с продвижением элемента start на один шаг. Мой вариант использования - перебирать игроков в игре в покер, продвигая шайбу дилера вперед на одного игрока в каждом раунде.
Johan
3

Если вы хотите изменитьn время цикла , ncycles примените рецепт itertools :

from itertools import chain, repeat


def ncycles(iterable, n):
    "Returns the sequence elements n times"
    return chain.from_iterable(repeat(tuple(iterable), n))


list(ncycles(["a", "b", "c"], 3))
# ['a', 'b', 'c', 'a', 'b', 'c', 'a', 'b', 'c']
пиланг
источник
2

Вам нужен собственный итератор - я адаптирую итератор из этого ответа .

from itertools import cycle

class ConnectionPool():
    def __init__(self, ...):
        # whatever is appropriate here to initilize
        # your data
        self.pool = cycle([blah, blah, etc])
    def __iter__(self):
        return self
    def __next__(self):
        for connection in self.pool:
            if connection.is_available:  # or however you spell it
                return connection
Итан Фурман
источник