Trình lặp danh sách tròn trong Python

Tôi cần lặp lại danh sách vòng tròn, có thể nhiều lần, mỗi lần bắt đầu với mục được truy cập cuối cùng.

Trường hợp sử dụng là một nhóm kết nối. Một ứng dụng khách yêu cầu kết nối, một trình lặp kiểm tra xem kết nối trỏ đến có sẵn không và trả về kết nối đó, nếu không sẽ lặp lại cho đến khi tìm thấy kết nối khả dụng.

Có cách nào gọn gàng để làm điều đó bằng Python không?

Sử dụng itertools.cycle, đó là mục đích chính xác của nó:

from itertools import cycle

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

pool = cycle(lst)

for item in pool:
    print item,

Đầu ra:

a b c a b c ...

(Vòng lặp mãi mãi, hiển nhiên)

Để nâng cấp trình vòng lặp theo cách thủ công và lấy từng giá trị từ nó, chỉ cần gọi next(pool):

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

Câu trả lời đúng là sử dụng itertools.cycle . Nhưng, giả sử rằng hàm thư viện không tồn tại. Bạn sẽ thực hiện nó như thế nào?

Sử dụng máy phát điện :

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

Sau đó, bạn có thể sử dụng một forcâu lệnh để lặp vô hạn hoặc bạn có thể gọi next()để nhận một giá trị tiếp theo duy nhất từ ​​trình lặp của trình tạo:

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

Hoặc bạn có thể làm như sau:

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

in abcdefab c ... mãi mãi

bạn có thể thực hiện điều này với append(pop())vòng lặp:

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

hoặc for i in range()vòng lặp:

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

hoặc đơn giản:

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

while 1:
    for i in l:
       print i

tất cả đều in:

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

trong ba cách tiếp cận, tôi sẽ có xu hướng tiếp cận append (pop ()) như một hàm

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]

Bạn cần một trình lặp tùy chỉnh - Tôi sẽ điều chỉnh trình lặp từ câu trả lời này .

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

Nếu bạn muốn quay vòng nthời gian, hãy triển khai ncycles công thức 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']