[Python3] 순열과 조합 (itertools permutations 사용법)

일반적인 순열과 조합 (저수준)

def permute(arr):
    result = [arr[:]]
    c = [0] * len(arr)
    i = 0
    while i < len(arr):
        if c[i] < i:
            if i % 2 == 0:
                arr[0], arr[i] = arr[i], arr[0]
            else:
                arr[c[i]], arr[i] = arr[i], arr[c[i]]
            result.append(arr[:])
            c[i] += 1
            i = 0
        else:
            c[i] = 0
            i += 1
    return result

 

위와 같은 방법은 저수준 방식으로 C/C++등 에서 작성하는 방식이다. (STL제외) 

 

파이썬에서는 복잡한 코드를 정의하는 함수를 사용하여 간단하게 만들 수 있다.

 

순열과 조합 in Python

import itertools

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

 

A, B, C

 

3개의 원소로 수열을 만들게 되면

 

'ABC', 'ACB', 'BAC', 'BCA', 'CAB', 'CBA'

 

3개의 원소로 이루어진 조합이 완성된다. 

 

번외로 permutations의 두번째 인자에 리스트의 크기보다 작은 수를 지정하여 다른 조합을 만들 수 있다.

 

예를 들어, 2를 넣으면

 

'AB', 'AC', 'BA', 'BC', 'CA', 'CB'

 

2개의 원소로 이루어진 조합이 완성된다.