[파이썬] 리스트 경우의 수, 조합 구하기

리스트에서 두 개의 원소로 할 수 있는 조합을 구하라는 문제와 비슷한 걸 자주 마주칠 수 있는데요 그럴 때 itertools를 사용하면 쉽게 풀 수 있습니다.

from itertools import product
from itertools import permutations
from itertools import combinations

 

 

하나의 리스트에서 구할 때는 permutations과 combinations를 사용하면 되고 두 개 이상의 리스트에서 구할 때는 product를 사용하면 됩니다. permutations은 정확히는 순열이 되고 combinations는 조합이 되겠네요

 

---

 

a = [1,2,3]
list(permutations(a, 2))

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

순열이기에 결과값에 순서가 영향이 있을 때 사용하시면 됩니다.

 

a = [1,2,3]
list(combinations(a,2))

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

조합은 순서가 영향력이 없을 때 사용하시면 됩니다.

 

---

 

from itertools import product
a = [[1,2,3], ['a','b'], ['!', '?']]
list(product(*a))

# [(1, 'a', '!'),
# (1, 'a', '?'),
# (1, 'b', '!'),
# (1, 'b', '?'),
# (2, 'a', '!'),
# (2, 'a', '?'),
# (2, 'b', '!'),
# (2, 'b', '?'),
# (3, 'a', '!'),
# (3, 'a', '?'),
# (3, 'b', '!'),
# (3, 'b', '?')]

product는 여러 리스트에서 추출할 때 사용됩니다.