본문 바로가기

Language/Python

python permutations

사용법

itertools 라이브러리 안에 있는 기능

from itertools import permutations
a = [1,2,3]
p = list(permutations(a,2))
# 2번째 인자부분은, 얻고자 하는 경우의 수의 범위 ex) 2는 (1,2), (1,3), (2,1), (2,3), (3,1), (3,2)가 된다.

근데 이렇게 하면, (1,2) (2,1) 처럼 같은 경우의 수가 불필요하게 반복되는 것을 볼 수 있다.

 

set_p = list(set([tuple(sorted(list(i))) for i in p]))
# 이렇게 하면, [(1,2), (1,3), (2,3)]이 출력된다.
반응형