生成列表清單的交叉產品

Question

我有一個Python列表，如[[1 2 3] [4 5] [3] [1]]，我想生成所有可能的排列。輸出應該是這樣的

[[1 4 3 1] [1 5 3 1] [2 4 3 1] [2 5 3 1] [3 4 3 1] [3 5 3 1]]

我想過，但循環嵌套循環的數量將根據所輸入的不同而不同。有人可以提出一個好的算法嗎？

Answer 1

你想要的是一個carthesian product。 itertoolshas a function product，它正是：

import itertools 
a = [[1, 2, 3], [4, 5], [3], [1]] 
p = itertools.product(*a) 
print list(p) 

此輸出

[(1, 4, 3, 1), 
(1, 5, 3, 1), 
(2, 4, 3, 1), 
(2, 5, 3, 1), 
(3, 4, 3, 1), 
(3, 5, 3, 1)]