itertools — 반복을 조립하는 표준 도구

import itertools as it

# count — range 인데 무한
for n in it.count(10, 2):
    if n > 20:
        break
    print(n)              # 10 12 14 16 18 20

# cycle — 라운드로빈
colors = it.cycle(["red", "green", "blue"])
for _ in range(7):
    print(next(colors))
# red green blue red green blue red

# repeat — 제한 또는 무한
print(list(it.repeat("x", 4)))    # ['x', 'x', 'x', 'x']

# 무한 zip + 유한 list
pairs = list(zip(it.count(1), ["a", "b", "c"]))
print(pairs)                       # [(1, 'a'), (2, 'b'), (3, 'c')]

import itertools as it

# chain — iterable 이어붙임
result = list(it.chain([1, 2], [3, 4], [5]))
print(result)                      # [1, 2, 3, 4, 5]

# zip_longest — 가장 짧은 거에서 안 멈춤
result = list(it.zip_longest([1, 2, 3], ["a", "b"], fillvalue="?"))
print(result)                      # [(1, 'a'), (2, 'b'), (3, '?')]

# islice — *모든* iterable 슬라이싱, 무한 포함
def naturals():
    n = 1
    while True:
        yield n
        n += 1

first_5 = list(it.islice(naturals(), 5))
print(first_5)                     # [1, 2, 3, 4, 5]

import itertools as it

items = ["apple", "ant", "banana", "berry", "apricot"]

# 순진한 groupby — 연속 매치만 그룹
for key, group in it.groupby(items, key=lambda x: x[0]):
    print(key, list(group))
# a ['apple', 'ant']
# b ['banana', 'berry']
# a ['apricot']               <- 두 번째 'a' 그룹, 합쳐지지 *않음*

# SQL 스타일 그룹핑은 정렬 먼저
items.sort(key=lambda x: x[0])
for key, group in it.groupby(items, key=lambda x: x[0]):
    print(key, list(group))
# a ['apple', 'ant', 'apricot']
# b ['banana', 'berry']

import itertools as it

# product — 데카르트 곱 (nested for 루프 대체)
for x, y in it.product([1, 2], ["a", "b"]):
    print(x, y)
# 1 a
# 1 b
# 2 a
# 2 b

# permutations — 모든 순서 있는 배치
print(list(it.permutations("abc", 2)))
# [('a', 'b'), ('a', 'c'), ('b', 'a'), ('b', 'c'), ('c', 'a'), ('c', 'b')]

# combinations — 순서 없는 선택
print(list(it.combinations("abc", 2)))
# [('a', 'b'), ('a', 'c'), ('b', 'c')]

# combinations_with_replacement — 재사용 가능한 선택
print(list(it.combinations_with_replacement("ab", 2)))
# [('a', 'a'), ('a', 'b'), ('b', 'b')]

import itertools as it

source = (x*x for x in range(5))
a, b, c = it.tee(source, 3)

print(list(a))               # [0, 1, 4, 9, 16]
print(list(b))               # [0, 1, 4, 9, 16]
print(list(c))               # [0, 1, 4, 9, 16]

# 주의 — tee() 후에는 원본 source를 다시 쓰지 마
# tee 버퍼에 들어감

# 대부분 경우 list 로 materialize 하는 게 더 명확
squares = [x*x for x in range(5)]