如何计算列表的所有组合

Question

我在Python问题上受阻，希望有人能帮助我。

问题实际上很简单。

我试图建立具有所有可能组合的列表，但是列表的元素范围不相同。

这是我的代码，我尝试使用for循环进行操作，但不起作用。

for j in range(0,size):


    for k, val in enumerate(self.Algo.Inputs[j].Values):
        self.Commandlist[j] = k   

self.Commandlist是具有固定范围的列表，并且首先填充零。

self.Commandlist = [0,0,0]

self.Algo.Inputs [j] .Values给我每个元素的大小，例如，如果self.Algo.Inputs [0] .Values = 4 self.Algo.Inputs [1] .Values = 1 self.Algo。输入[2]。值= 2

我想要所有组合，[0,0,0]，[1,0,0]，[2,0,0]，[3,0,0]，[4,0,0]，[0,1 ，0]，[1,1,0]，[2,1,0]，[3,1,0]，[4,1,0]等。

我想我忘了一个循环，但我不知道。 我也尝试了itertools模块的一些东西，但我无法使其工作。

感谢您的帮助。

Answer 1

如前所述，您可以使用itertools，例如：

import itertools
a = b = c = range(3) # you can specify different range for each one
[list(x) for x in list(itertools.product(a, b, c))]

结果：

[[0，0，0]，[0，0，1]，[0，0，2]，[0，1，0]，[0，1，1]，[0，1，2]，[ 0，2，0]，[0，2，1]，[0，2，2]，[1，0，0]，[1，0，1]，[1，0，2]，[1， 1，0]，[1、1、1]，[1、1、2]，[1、2、0]，[1、2、1]，[1、2、2]，[2、0， 0]，[2、0、1]，[2、0、2]，[2、1、0]，[2、1、1]，[2、1、2]，[2、2、0] ，[2、2、1]，[2、2、2]]