[Python] 순열과 조합

1. 순열

import itertools

pool = ['A', 'B', 'C']
print(list(map(''.join, itertools.permutations(pool))))

# 결과 : ['ABC', 'ACB', 'BAC', 'BCA', 'CAB', 'CBA']

2. 조합

import itertools

pool = ['A', 'B', 'C']
print(list(map(''.join, itertools.combinations(pool, 2))))

# 결과 : ['AB', 'AC', 'BC']

3. 중복 순열

import itertools

pool = ['A', 'B', 'C']
print(list(map(''.join, itertools.product(pool, repeat=2))))

# 결과 : ['AA', 'AB', 'AC', 'BA', 'BB', 'BC', 'CA', 'CB', 'CC']

4. 중복 조합

import itertools

pool = ['A', 'B', 'C']
print(list(map(''.join, itertools.combinations_with_replacement(pool, 2))))

# 결과 : ['AA', 'AB', 'AC', 'BB', 'BC', 'CC']

pool의 자료형으로는 리스트, 튜플, 딕셔너리, 문자열이 올 수 있다.