Iterator listy cyklicznej w Pythonie

Muszę powtarzać listę cykliczną, możliwie wiele razy, za każdym razem zaczynając od ostatnio odwiedzanego elementu.

Przypadek użycia to pula połączeń. Klient prosi o połączenie, iterator sprawdza, czy wskazane połączenie jest dostępne i zwraca je, w przeciwnym razie wykonuje pętlę, dopóki nie znajdzie takiego, które jest dostępne.

Czy jest na to zgrabny sposób w Pythonie?

Użyj itertools.cycle, to jest jego dokładny cel:

from itertools import cycle

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

pool = cycle(lst)

for item in pool:
    print item,

Wynik:

a b c a b c ...

(Oczywiście pętle na zawsze)

Aby ręcznie przesuwać iterator i wyciągać z niego wartości pojedynczo, po prostu wywołaj next(pool):

>>> next(pool)
'a'
>>> next(pool)
'b'

Prawidłowa odpowiedź to użycie itertools.cycle . Ale załóżmy, że funkcja biblioteki nie istnieje. Jak byś to zaimplementował?

Użyj generatora :

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

Następnie możesz użyć forinstrukcji, aby iterować w nieskończoność, lub możesz wywołać, next()aby uzyskać pojedynczą następną wartość z iteratora generatora:

connections = circular()
next(connections) # 'a'
next(connections) # 'b'
next(connections) # 'c'
next(connections) # 'a'
next(connections) # 'b'
next(connections) # 'c'
next(connections) # 'a'
#....

Lub możesz to zrobić:

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

wypisuje abcdefab c ... na zawsze

możesz to osiągnąć za pomocą append(pop())pętli:

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

lub for i in range()pętla:

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

lub po prostu:

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

while 1:
    for i in l:
       print i

z których wszystkie drukują:

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

z tych trzech byłbym podatny na podejście append (pop ()) jako funkcję

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]

Potrzebujesz niestandardowego iteratora - dostosuję iterator na podstawie tej odpowiedzi .

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

Jeśli chcesz zmieniać nczasy cykli , zastosuj ncycles przepis 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']