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)]'Language > Python' 카테고리의 다른 글
| 다차원 배열 복사할 때 효율적으로 하는 방법 (0) | 2021.09.09 |
|---|---|
| 8. 파이썬 거듭제곱 연산 (0) | 2021.08.27 |
| 7. 파이썬 math 라이브러리 (0) | 2021.08.12 |
| 6. 파이썬 input 읽기 - 백준 사이트에서 필수 사용! (0) | 2021.08.11 |
| 5. 파이썬 PriorityQueue vs heapq (0) | 2021.08.08 |
댓글