IT/python

[파이썬 모듈] 리스트의 모든 조합 구하는 방법

bs engineer 2022. 9. 21. 12:52

리스트의 모든 조합 구하는 방법에는 3가지가 있음

  1. product
  2. permutations
  3. combinations

하나의 리스트에서 모든 조합을 구할 때,

permutations, combinations 사용

from itertools import permutations
from itertools import combinations

items = ['a','b','c']
list(permutations(items,2)) # [('a','b'),('a','c'),('b','a'),('b','c'),('c,'a'),('c','b')]

list(combinations(items,2)) # [('a','b'),('a','c'),('b','c')]

두 개 이상의 리스트의 모든 조합 구할 때

from itertools import product

items = [['a','b'],['1'],['!','@']  
list(product(*items)) # [('a','1','!'],('a','1','@') ...]