[TIL] 250630 - 파이썬 itertools.product 사용법

• itertools.product는 중복을 허용하며, 주어진 리스트의 모든 가능한 조합(데카르트 곱)을 만들어주는 도구

기본 사용법

from itertools import product

for p in product(['A', 'B'], repeat=2):
    print(p)

출력 결과:

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

 

• repeat=2는 두 글자로 조합한다는 뜻
• 중복 허용
• 순서 고려함 → ('A', 'B')와 ('B', 'A')는 다르게 봄

사용 예시

🔍 'A', 'E', 'I', 'O', 'U'로 만들 수 있는 단어 생성

from itertools import product

alphabet = ['A', 'E', 'I', 'O', 'U']
all_words = []

for i in range(1, 6):  # 길이 1~5짜리 단어 생성
    for p in product(alphabet, repeat=i):
        all_words.append(''.join(p))