[Python] itertools 라이브러리

itertools 라이브러리를 사용하면 효율적으로 순열과 조합을 구현할 수 있다. 

permutations( ) - 순열

순열은 서로 다른 n개의 다른 원소를 가지는 어떤 집합에서 중복 없이 순서에 상관 있게 r개의 원소를 선택하거나 나열하는 것을 의미한다. 

import itertools
arr = ['A', 'B', 'C', 'D']
nPr = itertools.permutations(arr, 2)
print(list(nPr))
"""
[('A', 'B'), ('A', 'C'), ('A', 'D'), ('B', 'A'), 
('B', 'C'), ('B', 'D'), ('C', 'A'), ('C', 'B'), 
('C', 'D'), ('D', 'A'), ('D', 'B'), ('D', 'C')]
"""

combinations( ) - 조합

조합은 서로 다른 n개의 원소를 가지는 어떤 집합에서 중복 없이 순서에 상관 없이 r개의 원소를 선택하거나 나열하는 것을 의미한다. 

import itertools
arr = ['A', 'B', 'C', 'D']
nCr = itertools.combinations(arr, 2)
print(list(nCr))
## [('A', 'B'), ('A', 'C'), ('A', 'D'), ('B', 'C'), ('B', 'D'), ('C', 'D')]

 

product( ) - 중복 순열

서로 다른 n개의 원소에 대하여 중복을 허용하여 r개를 뽑아 순서 있게 나열하는 것을 의미한다.

import itertools
arr = ['A', 'B', 'C', 'D']
a = itertools.product(arr, repeat=2)
print(list(a))
"""
[('A', 'A'), ('A', 'B'), ('A', 'C'), ('A', 'D'), 
('B', 'A'), ('B', 'B'), ('B', 'C'), ('B', 'D'), 
('C', 'A'), ('C', 'B'), ('C', 'C'), ('C', 'D'), 
('D', 'A'), ('D', 'B'), ('D', 'C'), ('D', 'D')]
"""

 

combinations_with_replacement( ) - 중복 조합

서로 다른 n개의 원소를 가지는 어떤 집합에서 중복에 상관 없이, 순서에 상관 없이 r개의 원소를 선택

import itertools
arr = ['A', 'B', 'C', 'D']
a = itertools.combinations_with_replacement(arr, 2)
print(list(a))
"""
[('A', 'A'), ('A', 'B'), ('A', 'C'), ('A', 'D'), 
('B', 'B'), ('B', 'C'), ('B', 'D'), 
('C', 'C'), ('C', 'D'), 
('D', 'D')]
"""