본문 바로가기
Python/알고리즘팁

파이썬 순열/조합 permutation, combination 사용하기

by 붕어사랑 티스토리 2021. 4. 15.
반응형
from itertools import permutations, combinations
a = [1,2,3,4,5]

list(permutations(a,2))
결과 : [(1, 2), (1, 3), (1, 4), (2, 1), (2, 3), (2, 4), (3, 1), (3, 2), (3, 4), (4, 1), (4, 2), (4, 3)]


list(combinations(a,3))
결과 : [(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]

 

 

고등학교때 nPr  nCr 을 기억하고 있을 것이다.

 

n개에서 r개를 선택하여 나열하는 경우의수

n개에서 r개를 선택하는 경우의 수

 

permutations과 combinations는 위 두개를 계산해주는 함수이다.

 

permitations(인풋, r값)

combinations(인풋, r값)

 

이렇게 사용해주면 된다.

반응형

댓글