Có một cách thanh lịch để chu kỳ thông qua một danh sách N lần thông qua lặp (như itertools.cycle nhưng giới hạn các chu kỳ)?

Question

Tôi muốn chuyển qua danh sách nhiều lần (N lần) qua một trình lặp, để không thực sự lưu trữ N bản sao của danh sách trong bộ nhớ. Có cách xây dựng hoặc thanh lịch để thực hiện việc này mà không cần viết trình tạo của riêng tôi không?

Lý tưởng nhất, itertools.cycle (my_list) sẽ có đối số thứ hai để giới hạn số lần nó lặp lại ... than ôi, không may mắn như vậy.

Answer 1

import itertools 
itertools.chain.from_iterable(itertools.repeat([1, 2, 3], 5)) 

Itertools là một thư viện tuyệt vời. :)

Answer 2

itertools.chain.from_iterable(iter(L) for x in range(N)) 

Answer 3

Đối với trường hợp đặc biệt mà bạn cần phải lặp qua danh sách nhiều lần, điều này không quá tệ.

Nó tạo ra một danh sách các tài liệu tham khảo n-my_list, vì vậy nếu n là rất lớn nó là tốt hơn để sử dụng câu trả lời Darthfelt của

>>> import itertools as it 
>>> it.chain(*[my_list]*n) 

Answer 4

Tất cả các câu trả lời khác là tuyệt vời. Một giải pháp khác là sử dụng islice. Điều này cho phép bạn làm gián đoạn chu kỳ tại bất kỳ điểm:

>>> from itertools import islice, cycle 
>>> l = [1, 2, 3] 
>>> list(islice(cycle(l), len(l) * 3)) 
[1, 2, 3, 1, 2, 3, 1, 2, 3] 
>>> list(islice(cycle(l), 7)) 
[1, 2, 3, 1, 2, 3, 1] 

Answer 5

Bạn nói rằng bạn không muốn viết máy phát điện riêng của bạn, nhưng một biểu thức máy phát điện có lẽ sẽ là cách dễ nhất và hiệu quả nhất để thực hiện những gì bạn' sau đó. Nó không yêu cầu bất kỳ cuộc gọi chức năng hoặc nhập khẩu của bất kỳ mô-đun. itertools là một mô-đun tuyệt vời, nhưng có lẽ không cần thiết trong trường hợp này?

some_list = [1, 2, 3] 
cycles = 3 
gen_expr = (elem for _ in xrange(cycles) for elem in some_list) 

hoặc chỉ

(elem for _ in xrange(3) for elem in [1, 2, 3]) 

hoặc

for elem in (e for _ in xrange(3) for e in [1, 2, 3]): 
    print "hoo-ray, {}!".format(elem) 

Answer 6

@ câu trả lời Darthfett được ghi nhận như một itertools recipes:

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"], 3)) 
# ['a', 'b', 'a', 'b', 'a', 'b'] 

Để thuận tiện, tôi thêm rằng các thư viện more_itertools thực hiện công thức này (và nhiều công thức khác) cho bạn:

import more_itertools as mit 

list(mit.ncycles(["a", "b"], 3)) 
# ['a', 'b', 'a', 'b', 'a', 'b']