순열과 조합 구하기 (Permutation & Combination)

파이썬에서 순열과 조합의 경우를 구하는 방법

itertools를 이용하자!

 

Permutation (순열)

import itertools

for perm in itertools.permutations(['a','b','c']):
    print(perm)
    
# ===
('a', 'b', 'c')
('a', 'c', 'b')
('b', 'a', 'c')
('b', 'c', 'a')
('c', 'a', 'b')
('c', 'b', 'a')

Combination (조합) -3C2

[1,2,3] 에서 2개를 가지고 조합을 만들때, itertools.combinations(list, <만들 조합원소수>)

import itertools

for comb in itertools.combinations(['a','b','c'],2):
    print(comb)

# ===
('a', 'b')
('a', 'c')
('b', 'c')