可变长度嵌套循环Python

Question

我正在尝试在 python 中嵌套一个可变长度的 for 循环并能够重用这些变量。

for i in range(0,256):
    for j in range(0,256):
        for k in range(0,256):
            ...
            myvar[i][j][k]...

在上面的代码中，有一个长度为 3 的硬编码嵌套 for 循环。

它应该像这样工作：

for index[0] in range(0,256):
  for index[1] in range(0,256):
    for index[2] in range(0,256):
      ...
        for index[n-1] in range(0,256):
          for index[n] in range(0,256):
            myvar[index[0]][index[1]][index[2]] ... [index[n-1]][index[n]] ...

对于n的任意值

Answer 1

一种方法是对值的排列进行单个循环。

考虑itertools.product的这种行为：

>>> list(product(*[range(3)] * 2))
[(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]

因此，您可以在一些迭代器的乘积上仅在一个循环中从一堆嵌套循环中获得相同的值：

from itertools import product

number = 3
ranges = [range(256)] * number
for values in product(*ranges):
    ...