嵌套列表的组合

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')]