[TIL] 250529 - 파이썬 itertools.permutations 사용법

파이썬에서 순열을 구하려 할 때, itertools 모듈의 permutations 함수를 사용하면 아주 간단하게 구현할 수 있습니다. 이번 글에서는 permutations 함수의 사용법과 예제를 간단히 정리해보았습니다. 

 

1. permutations 기본 사용법

from itertools import permutations

data = ['A', 'B', 'C']
result = permutations(data, 2)

for item in result:
    print(item)

 

출력 결과 ⬇️

('A', 'B')
('A', 'C')
('B', 'A')
('B', 'C')
('C', 'A')
('C', 'B')

• permutations(data, r)은 data에서 r개를 뽑아 순서를 고려한 조합(순열)을 튜플 형태로 반환합니다.

 

2. 모든 자리수의 순열 생성 (1자리, 2자리, ..., n자리)

위의 내용을 활용해 문자열 '123'으로 만들 수 있는 모든 자리수의 순열 조합을 구해보도록 하겠습니다.

from itertools import permutations

s = "123"
result = set()

for i in range(1, len(s)+1):
    for p in permutations(s, i):
        result.add(''.join(p))

print(result)

 

출력 결과 ⬇️

{'123', '1', '12', '213', '21', '231', '132', '13', '2', '3', '31', '32', '312', '321'}