可变长度嵌套循环Python

问题描述

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

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

在上面的代码中，有一个长度为3for的硬编码嵌套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

---

解决方案

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

考虑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):
    ...