生成阵列列的所有可能的组合在Python

Question

例如，我有数组这样

[ 
    [1,2,3], 
    [4], 
    [5,6], 
] 

我要生成从上面的列表，这是否应该是这个样子的所有组合。

[1, 4, 5] 
[1, 4, 6] 
[2, 4, 5] 
[2, 4, 6] 
[3, 4, 5] 
[3, 4, 6] 

Answer 1

听起来像是你想从product内置itertools库

>>> import itertools 
>>> list(itertools.product([1, 2, 3], [4], [5, 6])) 
[(1, 4, 5), (1, 4, 6), (2, 4, 5), (2, 4, 6), (3, 4, 5), (3, 4, 6)] 
>>> 
>>> columns = [[1,2,3], 
       [4], 
       [5,6]] 
>>> list(itertools.product(*columns)) 
[(1, 4, 5), (1, 4, 6), (2, 4, 5), (2, 4, 6), (3, 4, 5), (3, 4, 6)] 

Answer 2

在这里你去：

a = [1,2,3] 
b = [4] 
c = [5,6] 

d = [[x, y, z] for x in a for y in b for z in c] 

Answer 3

要进行笛卡尔乘积，你只需要在各个方面进行迭代在这种情况下，

例如：

for x in dimension_x: 
    for y in dimension_y: 
     for z in dimension_z: 
      use the x,y,z 

的算法的复杂性将始终是艰难（2个阵列 - > N2，3 - > N3，...，对于M - > N R个M其中n是最长的阵列的长度）。

请注意，您有重复项： （a，b）与（b，a）相同。因此，如果您不需要重复项，则可以更改算法以加快工作速度。