[python] 순열과 조합 - 라이브러리 사용

순열(Permutations)

=> 서로 다른 n개 중 r개를 뽑는 뒤 나열 하는 것

array = [1, 2, 3, 4] # n 원소 개수
r = 2                # r 뽑는 개수  

from itertools import permutations

per = list(permutations(array,r))
print(per)

[(1, 2), (1, 3), (1, 4), (2, 1), (2, 3), (2, 4), (3, 1), (3, 2), (3, 4), (4, 1), (4, 2), (4, 3)]

 

중복순열

=> '순열'에서 중복을 포함하는 경우

array = [1, 2, 3, 4] # n 원소 개수
r = 2                # r 뽑는 개수  

from itertools import product

per = list(product(array,repeat = r))
print(per)

[(1, 1), (1, 2), (1, 3), (1, 4), (2, 1), (2, 2), (2, 3), (2, 4), (3, 1), (3, 2), (3, 3), (3, 4), (4, 1), (4, 2), (4, 3), (4, 4)]

 

조합(Combinations)

=> 서로 다른 n개 중 r래를 뽑는 것 (순서를 생각하지 않는다)

array = [1, 2, 3, 4] # n 원소 개수
r = 2                # r 뽑는 개수  

from itertools import combinations

per = list(combinations(array,r))
print(per)

[(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]    

 

중복조합

=> '조합'에서 중복을 포함하는 경우

array = [1, 2, 3, 4] # n 원소 개수
r = 2                # r 뽑는 개수  

from itertools import combinations_with_replacement

per = list(combinations_with_replacement(array,r))
print(per)

[(1, 1), (1, 2), (1, 3), (1, 4), (2, 2), (2, 3), (2, 4), (3, 3), (3, 4), (4, 4)]