如何排列/組合不同大小的列表？ python

Question

我有一個列表清單，沒有。內部列表的是未知：

>>> x = [1,2,3] 
>>> y = [4,5,6,7] 
>>> z = [8] 
>>> lol = [x,y,z] 

我需要在lol列表的順序內部列表內得到每個項目的組合，我一直在做它是這樣來獲得所需的輸出：

>>> for i in x: 
...  for j in y: 
...    for k in z: 
...      print [i,j,k] 
... 
[1, 4, 8] 
[1, 5, 8] 
[1, 6, 8] 
[1, 7, 8] 
[2, 4, 8] 
[2, 5, 8] 
[2, 6, 8] 
[2, 7, 8] 
[3, 4, 8] 
[3, 5, 8] 
[3, 6, 8] 
[3, 7, 8] 

什麼是pythonic方式做以上？這是否有itertools函數？

我已經試過itertools.product但我沒有得到期望的輸出：

>>> from itertools import product 
>>> for i in product(lol): 
...  print i 
... 
([1, 2, 3],) 
([4, 5, 6, 7],) 
([8],) 

Answer 1

你是非常接近：

>>> for i in product(*lol): 
... print i 
... 
(1, 4, 8) 
(1, 5, 8) 
(1, 6, 8) 
(1, 7, 8) 
(2, 4, 8) 
(2, 5, 8) 
(2, 6, 8) 
(2, 7, 8) 
(3, 4, 8) 
(3, 5, 8) 
(3, 6, 8) 
(3, 7, 8) 

在doc，有喜歡product(A, B)例子。所以你應該通過每個列表作爲參數。在你的情況下，product(x,y,z)會做到這一點。 *lol表示法用於unpacking。

Answer 2

itertools.product爲每個集合取一個單獨的參數，而不是一個包含集合的單個迭代。

for i in product(*lol): 
    print i 

Answer 3

如其他答案建議，您需要解壓lol。 並作爲替代，列表理解是更Python在我看來：

import itertools 


x = [1,2,3] 
y = [4,5,6,7] 
z = [8] 
lol = [x,y,z] 

print [each for each in itertools.product(*lol)] # or set() 

[(1, 4, 8), (1, 5, 8), (1, 6, 8), (1, 7, 8), (2, 4, 8), (2, 5, 8), (2, 6, 8), (2, 7, 8), (3, 4, 8), (3, 5, 8), (3, 6, 8), (3, 7, 8)]