[Python] itertools cycle()

itertools의 cycle()을 사용하여 원하는 수 만큼 리스트를 반복할 수 있습니다.

from itertools import cycle

test = [
    cycle([1, 2, 3, 4, 5]),
    cycle([1, 2, 3]),
    cycle([1, 2]),
]

for _ in range(10):
    print(str(next(test[0])) + ' ', end='')
print()

for _ in range(10):
    print(str(next(test[1])) + ' ', end='')
print()

for _ in range(10):
    print(str(next(test[2])) + ' ', end='')

# 출력결과
# 1 2 3 4 5 1 2 3 4 5 
# 1 2 3 1 2 3 1 2 3 1
# 1 2 1 2 1 2 1 2 1 2