是否有一种优雅的方式通过迭代遍历列表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']