嵌套列表的組合

Question

我有三個由其他函數生成的列表。假設現在他們是：

x = ['d', 'e'] 
g = ['1', '2'] 
y = ['f', g] 

正如你所看到的，g是y一部分。我正試圖獲得三個列表元素的所有組合。我試圖要對此有兩種方式：

方式一：

l = [] 
l.append([a]+[b] for a in x for b in y) 

另一種方法使用itertools：

import itertools 
l = list(itertools.product([a for a in x], [b for b in y])) 

這兩種方式產生以下組合：

[('d', 'f'), ('d', ['1', '2']), ('e', 'f'), ('e', ['1', '2'])] 

但我想得到的是：

[('d', 'f'), ('d', '1'), ('d','2'), ('e', 'f'), ('e', '1'), ('e','2')] 

另外，當例如x爲空時，我仍然沒有得到任何組合，當我仍然期望獲得其餘兩個列表的元素組合。

Answer 1

這是從@Psidoms答案的啓發而只是採用了特別定製flatten功能，以確保只有項目應該扁平化的迭代：

def flatten(x, types=list): 
    lst = [] 
    for item in x: 
     if isinstance(item, types): 
      for subitem in item: 
       lst.append(subitem) 
     else: 
      lst.append(item) 
    return lst 

>>> from itertools import product 

>>> list(product(x, flatten(y))) 
[('d', 'f'), ('d', '1'), ('d', '2'), ('e', 'f'), ('e', '1'), ('e', '2')] 

注意，有不幸的是在標準庫中無此flatten功能，但你可以也可以使用外部庫中的一個，例如iteration_utilities.deepflatten。注意，這需要提供str或basestring爲ignore：

>>> from iteration_utilities import deepflatten 

>>> list(product(x, deepflatten(y, ignore=str))) 
[('d', 'f'), ('d', '1'), ('d', '2'), ('e', 'f'), ('e', '1'), ('e', '2')] 

---

要排除空iterables從產品只是排除空subiterables。例如：

>>> x = [] 
>>> iterables = [subiterable for subiterable in (x, list(deepflatten(y, ignore=str))) if subiterable] 
>>> list(product(*iterables)) 
[('f',), ('1',), ('2',)] 

Answer 2

由於@BrenBarn評論，您可以拼合列表y與chain功能，然後用product：

from itertools import product, chain 

list(product(x, chain.from_iterable(y))) 
# [('d', 'f'), ('d', '1'), ('d', '2'), ('e', 'f'), ('e', '1'), ('e', '2')] 

Answer 3

我想指出兩種實現用於more_itertools可用平鋪狀的功能（通過pip install more_itertools安裝）。

flatten是itertools recipe和模擬@ Psidom的建議：

import itertools as it 
import more_itertools as mit 

list(it.product(x, mit.flatten(y))) 
# [('d', 'f'), ('d', '1'), ('d', '2'), ('e', 'f'), ('e', '1'), ('e', '2')] 

然而，扁平化更深層嵌套iterables，可以考慮使用collapse：

# Example 
x = ['d', 'e'] 
g = [('1'), [[['2']]]] 
y = [{'f'}, g] 

# Bad 
list(it.product(x, mit.flatten(y))) 
# [('d', 'f'), ('d', '1'), ('d', [[['2']]]), ('e', 'f'), ('e', '1'), ('e', [[['2']]])] 

# Good 
list(it.product(x, mit.collapse(y))) 
# [('d', 'f'), ('d', '1'), ('d', '2'), ('e', 'f'), ('e', '1'), ('e', '2')]