파이썬 itertools로 반복문을 더 강력하게: cycle, chain, combinations 활용법

파이썬 itertools로 반복문을 더 강력하게: cycle, chain, combinations 활용법 🔁

itertools는 반복을 더 효율적으로 처리할 수 있도록 도와주는 파이썬 표준 라이브러리입니다. 그 중에서도 자주 쓰이는 cycle(), chain(), combinations() 함수에 대해 실전 예제로 알아봅니다.

✅ cycle(): 무한 반복 순환

리스트의 요소를 무한히 순환할 수 있습니다. 예: 로테이션 기능

from itertools import cycle

colors = ['red', 'green', 'blue']
cycler = cycle(colors)

for _ in range(6):
    print(next(cycler))

✅ chain(): 여러 리스트 연결

여러 개의 리스트를 하나로 연결하여 반복할 수 있습니다.

from itertools import chain

a = [1, 2]
b = [3, 4]
c = [5]

combined = list(chain(a, b, c))
print(combined)

✅ combinations(): 조합 생성

리스트에서 가능한 조합을 생성할 수 있습니다. 예: 조합 문제 해결

from itertools import combinations

items = ['A', 'B', 'C']
for combo in combinations(items, 2):
    print(combo)

itertools를 사용하면 복잡한 반복문 로직도 훨씬 간결하고 효율적으로 만들 수 있어요. 데이터 처리나 알고리즘 문제를 다룰 때 큰 도움이 됩니다.

반복을 단순히 반복하지 마세요. itertools로 반복을 설계하세요. 작은 차이가 더 나은 개발자를 만듭니다. 🚀

Icons by Flaticon