[Python3] 모든 경우의 수, 곱집합 구하기 (itertools product)

iterable1 = 'ABCD'
iterable2 = 'xy'
iterable3 = '1234'

for i in iterable1:
    for j in iterable2:
        for k in iterable3:
            print(i+j+k)

 

a, b, c, d

x, y

1, 2, 3, 4 

 

위 소스코드는 3개의 리스트의 각 원소에 대해 곱집합을 구하는 소스코드이다.

 

ax1, ax2, ... , cy3, cy4, ... , dy4

 

위와 같이 모든 경우를 구하려면 3중 반복문을 통해 해결이 가능하다. 하지만, 퍼포먼스 측면에서 좋지 않을 뿐더러 소스코드의 가독성이 떨어질 수 있다.

 

import itertools

iterable1 = 'ABCD'
iterable2 = 'xy'
iterable3 = '1234'
itertools.product(iterable1, iterable2, iterable3)

 

itertools.product를 사용하면 모든 경우의 수 또는 곱집합을 구할 수 있다.