python 순열 = itertools.permutations

해당 내용은 programmers의 "파이썬을 파이썬답게"라는 강의를 보고 개인적인 공부를 위해 기록한 것입니다.

 

import itertools

pool = ['A', 'B', 'C']
print(list(map(''.join, itertools.permutations(pool)))) # 3개의 원소로 수열 만들기
print(list(map(''.join, itertools.permutations(pool, 2)))) # 2개의 원소로 수열 만들기

주의 ! 위의 함수는 사전 순서의 순열이 아닌 주어진 리스트의 순열을 구하는 함수이다. 따라서 사전 순서대로 순열을 구하고 싶으면 사전에 정렬을 해야함

 

깨달은 점

파이썬은 모든 것이 객체이며, class이기 때문에 어떤 값을 반환하는 함수 역시 해당 class 타입으로 반환할 수 있다. 따라서 기본 타입으로 형변환을 하는 것이 중요하다

ex) map(), zip(), itertools.permutations(), itertools.product