是否有一種優雅的方式通過迭代遍歷列表N次（如itertools.cycle但限制循環）？

Question

我想通過一個迭代器反覆遍歷一個列表（N次），以免在列表中實際存儲N個列表副本。有沒有內置或優雅的方式來做到這一點，而不寫我自己的發電機？

理想情況下，itertools.cycle（my_list）將有第二個參數來限制它循環的次數...唉，沒有這樣的運氣。

Answer 1

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

Itertools是一個奇妙的圖書館。 :)

Answer 2

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

Answer 3

對於需要多次迭代列表的特殊情況，這並不算太壞。

它確實創造n引用列表來my_list，所以如果n是非常大的，最好是用Darthfelt的回答

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

Answer 4

其他所有的答案都非常優秀。另一種解決方案是使用islice。這使您可以在任何時候中斷週期：

>>> 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

你說你不想寫自己的發電機，但發電機表達式很可能會完成你」最簡單，最有效的方法重新過後。它不需要任何函數調用或任何模塊的導入。 itertools是一個很好的模塊，但在這種情況下可能不需要？

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

或只是

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

或

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

Answer 6

@ Darthfett的答案被記錄爲一個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'] 

爲了方便，我補充一點，more_itertools庫實現這個配方（和許多其他）爲你：

import more_itertools as mit 

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