순열/조합 - itertools 라이브러리 사용

분류

모듈

속성

생성일

2022/01/27 20:21

최종 수정

2022/01/27 21:19

순열

itertools.permutations(iterable, r=None)

첫 번째 인자에는 iterable 객체가 오고, 두 번째 인자 r 은 iterable 객체 전체에서 순열을 뽑는 개수(r개) 를 지정한다. r 값은 디폴트로 None 이며, 이 때는 iterable 객체 전체에서 순열을 돌린다.

순열 튜플들을 리턴한다.

from itertools import permutations

l = ['A', 'B', 'C']

for i in permutations(l): #r을 지정하지 않거나 r=None으로 하면 최대 길이의 순열이 리턴된다!
	print(i) # permutations() 은 튜플들을 리턴한다.

'''
출력결과:
('A', 'B', 'C')
('A', 'C', 'B')
('B', 'A', 'C')
('B', 'C', 'A')
('C', 'A', 'B')
('C', 'B', 'A')
'''
Python
복사

items = [1, 2, 3, 4]

print(list(permutations(items, 2)))
# [(1, 2), (1, 3), (1, 4), (2, 1), (2, 3), (2, 4), (3, 1), (3, 2), (3, 4), (4, 1), (4, 2), (4, 3)]
Python
복사

조합

itertools.combinations(iterable, r)

조합에서는 두 번째 인자인 r 이 필수다.

from itertools import combinations

l = [1,2,3]

for i in combinations(l,2):
	print(i)

'''
출력 결과:
(1, 2)
(1, 3)
(2, 3)
'''
Python
복사

items = [1, 2, 3, 4]

print(list(combinations(items, 2)))
# [(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]
Python
복사

중복순열

itertools.product(*iterable, repeat=1)

순열, 조합과는 다르게, 여러 iterable 을 인자로 넣어줄 수 있고, iterable 들에 원소들을 하나씩 빼와서 순서쌍을 구성한다. (중복순열)

from itertools import product

l1 = ['A', 'B']
l2 = ['1', '2']

for i in product(l1,l2,repeat=1): #l1과 l2에서 각각 1개씩을 뽑은 중복순열 순서쌍
	print(i)

'''
출력결과:
('A', '1')
('A', '2')
('B', '1')
('B', '2')
'''

for i in product(l1,repeat=3): #li에서 중복을 허용해서 3개씩을 뽑은 순서쌍 
	# cf. product(l1,l1,l1,repeat=1)과 동일한 출력
	print(i)

'''
출력결과:
('A', 'A', 'A')
('A', 'A', 'B')
('A', 'B', 'A')
('A', 'B', 'B')
('B', 'A', 'A')
('B', 'A', 'B')
('B', 'B', 'A')
('B', 'B', 'B')
'''
Python
복사

중복조합

combinations_with_replacement(iterable,r)

r 은 필수인자이다.

iterable 에서 원소 개수가 r 개인 중복조합 튜플들을 리턴한다.

from itertools import combinations_with_replacement

l = ['A', 'B', 'C']

for i in combinations_with_replacement(l,2):
	print(i)

'''
출력결과:
('A', 'A')
('A', 'B')
('A', 'C')
('B', 'B')
('B', 'C')
('C', 'C')
'''
Python
복사