python：我如何實現列表中所有列表的笛卡爾積？

Question

特別是我正在查看是否有一些cartesian_product（）方法內置到python中，或者使用itertools來完成這項工作，從而避免編寫嵌套的多維循環。

例如我有以下

input = [ 
    [1], 
    [[2]], 
    [3], 
    [[a,b,c] , [[z,x,y]] , [d,f,g]], 
    4 
] 

期待：

output = [ 
    [1,2,3,a,b,c,4] 
    [1,2,3,z,x,y,4] 
    [1,2,3,d,f,g,4] 
] 

的一個特別的挑戰是，如在上述輸入列表，一個項目可以具有一個列表的任何層，但它應該能夠忽略所有這些並且仍然產生平坦的結果。

Answer 1

要添加上@ rorra的回答的頂部，以打印結果「扁平」，添加以下功能：

def flat(l): 
    def _flat(l, r):  
     if type(l) is not list: 
      r.append(l) 
     else: 
      for i in l: 
       r = r + flat(i) 
     return r 
    return _flat(l, []) 

，然後（rorra的有輕微的修改代碼）：

_input = [ 
    [1], 
    [[2]], 
    [3], 
    [['a','b','c'] , [['z','x','y']], ['d','f','g']], 
    4 
] 

_input = map(lambda(x): x if isinstance(x, list) else [x], _input) 

import itertools 
for element in itertools.product(*_input): 
    print flat(list(element)) # this line was modified 

輸出

[1, 2, 3, 'a', 'b', 'c', 4] 
[1, 2, 3, 'z', 'x', 'y', 4] 
[1, 2, 3, 'd', 'f', 'g', 4] 

Answer 2

input = [ 
    [1], 
    [[2]], 
    [3], 
    [['a','b','c'] , [['z','x','y']], ['d','f','g']], 
    4 
] 

input = map(lambda(x): x if isinstance(x, list) else [x], input) 

import itertools 
for element in itertools.product(*input): 
    print element