Python: itertools 활용하기

permutations

from itertools import permutations
nums = [num + 1 for num in range(3)]

for cases in permutations(nums, 2):
    print(cases, end=' ')

• permutations(iterater, reapeat)를 통해 순열을 구할 수 있다.
• 결과 : (1, 2) (1, 3) (2, 1) (2, 3) (3, 1) (3, 2), 가능한 모든 순서로 반복을 하지 않음

combinations

from itertools import combinations
nums = [num + 1 for num in range(3)]

for cases in combinations(nums, 2):
    print(cases, end=' ')

• combinations(iterator, repeat)를 통해 순열을 구할 수 있다.
• 결과 : (1, 2) (1, 3) (2, 3), 정렬된 순서로 반복을 하지 않음

product

from itertools import product
nums = [num + 1 for num in range(3)]

for cases in product(nums, repeat=2):
    print(cases, end=' ')

• product(iterator, iterator) 또는 product(iteratore, repeat=)로 순열을 구할 수 있다.
• 결과 : (1, 1) (1, 2) (1, 3) (2, 1) (2, 2) (2, 3) (3, 1) (3, 2) (3, 3), permuation에서 (1, 1), (2, 2), (3, 3)이 추가된 것을 알 수 있다. 즉 데카르트의 곱이다.

combinations_with_replacement

from itertools import combinations_with_replacement
nums = [num + 1 for num in range(3)]

for cases in combinations_with_replacement(nums, 2):
    print(cases, end=' ')

• combinations_with_replacement(iterator, num) 으로 순열을 구할 수 있다.
• 결과 : (1, 1) (1, 2) (1, 3) (2, 2) (2, 3) (3, 3), combination에서 (1, 1), (2, 2), (3, 3)이 추가된 것을 알 수 있다.