在itertools中，如何在特定次数的迭代后打破循环？

Question

我有一个重复的循环，我希望它在重复5次后停止重复：

from itertools import cycle
a = [1,2,3]

for i in cycle(a):
    print (i)
    if i == 5:
       break

我所期望的是123,123,123,123,123。它在断裂前循环5次。相反，它会一直持续下去。在继续下一个代码之前，我如何让它只循环5次

Answer 1

您可以使用@Selcuk在其答案中提到的enumerate，也可以使用next步进迭代器，直到它达到给定的迭代次数

import itertools
n = 5

a = [1,2,3]
g = itertools.cycle(a)

for i in range(n*len(a)):
    print(next(g))

您还可以查看itertools.repeat()-

import itertools
n = 5

a = [1,2,3]
for i in itertools.repeat(a, 5):
    for j in i:
        print(j)