파이썬 코딩테스트에서 자주 사용되는 표준 라이브러리 (3) itertools

itertools 모듈

• 반복자 생성 및 조합, 순열 등을 생성할 때 사용합니다.

 

from itertools import permutations, combinations, product

# permutations 예시
data = ['a', 'b', 'c']
perm = list(permutations(data, 2))
print(perm)  # Output: [('a', 'b'), ('a', 'c'), ('b', 'a'), ('b', 'c'), ('c', 'a'), ('c', 'b')]

# combinations 예시
comb = list(combinations(data, 2))
print(comb)  # Output: [('a', 'b'), ('a', 'c'), ('b', 'c')]

# product 예시
prod = list(product(data, repeat=2))
print(prod)  # Output: [('a', 'a'), ('a', 'b'), ('a', 'c'), ('b', 'b'), ('b', 'c'), ('c', 'c')]