본문 바로가기
Language/Python

리스트 안의 리스트를 파라미터로 넣기 Asterisk

by 규나 2021. 8. 28.
SMALL

아래와 같은 리스트가 있다고 하자.

A = [[1,3], [2,4], [4,6]]

이 A리스트 안의 리스트들로 product 함수에 적용해서 조합을 만들고 싶다. 

ex ) n개의 주사위에 서로 다른 숫자들이 적혀있고, 각 주사위에서 나온 숫자들로 조합을 만들고 싶은 경우

 

리스트 A의 크기가 정해져있지 않다면, 굉장히 난감하다.

이럴 때에는 아래와 같이 표현하면 가능하다.

list(itertools.product(*A))
# [(1, 2, 4), (1, 2, 6), (1, 4, 4), (1, 4, 6), (3, 2, 4), (3, 2, 6), (3, 4, 4), (3, 4, 6)]

댓글