python itertools：如何掷 N 个骰子？ [复制]

Question

我可以使用如下循环遍历 1、2 或 3 个 6 面骰子的所有组合：

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 dice 执行此操作？

Answer 1

这就是itertools.product 的用途。

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