2016-09-08 3 views
0

나는이 같은 루프와 함께 1, 2 또는 3 6면 주사위의 모든 조합을 반복 할 수 있습니다python itertools : N 주사위를 굴리는 방법은 무엇입니까?

for a in range(1, 7): # one die 
    sum = a 
    dosomething(sum) 

for a in range(1, 7): # two dice 
    for b in range(1, 7): 
    sum = a + b 
    dosomething(sum) 

for a in range(1, 7): # three dice 
    for b in range(1, 7): 
    for c in range(1, 7): 
     sum = a + b + c 
     dosomething(sum) 

어떻게 itertools를 사용하여 N 주사위를 위해이 작업을 수행 할 수 있습니까?

답변

4

이것은 itertools.product을위한 것입니다.

for dice in itertools.product(range(1, 7), repeat=N): 
    dosomething(sum(dice))