[Python] iterator 표준 라이브러리

📦 itertools 주요 함수 요약 + 예제

✅ 예제 코드

1. count()

from itertools import count

for i in count(3, 2):  # 3부터 2씩 증가
    print(i)
    if i > 10:
        break

2. combinations()

from itertools import combinations

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

('A', 'B')
('A', 'C')
('B', 'C')

3. permutations()

from itertools import permutations

data = ['A', 'B', 'C']
for p in permutations(data, 2):
    print(p)

('A', 'B')
('A', 'C')
('B', 'A')
('B', 'C')
('C', 'A')
('C', 'B')

4. zip_longest() ✅

from itertools import zip_longest

a = [1, 2, 3]
b = ['a', 'b']

zipped = list(zip_longest(a, b, fillvalue='없음'))
print(zipped)

[(1, 'a'), (2, 'b'), (3, '없음')]

• 일반 zip()은 짧은 쪽 기준으로 멈추지만,

• zip_longest()는 긴 쪽 기준으로 끝까지 가고, 부족한 값은 fillvalue로 채웁니다.

🔚 정리

• itertools는 반복 작업을 간결하고 효율적으로 만들어 줍니다.

• 조합, 순열, 누적합, 무한 반복, 리스트 채우기 등 다양하게 활용 가능

• 특히 zip_longest()는 길이가 다른 리스트 병합에 매우 유용합니다.