可变长度嵌套循环Python

2019年8月4日 192次阅读

我试图在
python中嵌套for循环的变量长度,并能够重用变量.

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

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

它应该像这样工作：

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):
    ...
点赞