优雅的方式来获得迭代列表的组合

Question

我有一个可迭代对象的列表，我有兴趣获得由每个迭代组成的0或1个项目的所有列表（顺序不重要，所以它的组合不是排列组合我寻求）。

我有一个非常不雅的实现，我已经在下面发布。

我相信有一种更优雅的方式来做到这一点，可能与itertools模块，但我不能拿出任何东西。有什么建议？

---

import itertools 


def all_subsets(ss): 
    subset_lens = range(0, len(ss) + 1) 
    list_of_subsets = map(lambda n: itertools.combinations(ss, n), subset_lens) 
    return itertools.chain.from_iterable(list_of_subsets) 


list_of_iterables = [["A1"], ["B1", "B2", "B3"], ["C1", "C2"]] 

all_possibilities = itertools.chain.from_iterable(itertools.product(*subset) 
           for subset in all_subsets(list_of_iterables)) 

# Visual representation of the desired result 
for eg in all_possibilities: 
    print eg 

结果：

() 
('A1',) 
('B1',) 
('B2',) 
('B3',) 
('C1',) 
('C2',) 
('A1', 'B1') 
('A1', 'B2') 
('A1', 'B3') 
('A1', 'C1') 
... 

Answer 1

[filter(None, comb) for comb in itertools.product(*[[None] + it for it in list_of_iterables])] 

这使得一对夫妇简化假设。如果您的迭代器包含的值在布尔上下文中不是true，那么您必须使用更复杂的filter。如果您的iterables不是列表，则必须使用itertools.chain而不是[None] + it。

Answer 2

这就是我想出了...

data = [["A1"], ["B1", "B2", "B3"], ["C1", "C2"]] 
data = [[None] + x for x in data] 
data = sorted(filter(None, x) for x in itertools.product(*data)) 
for result in data: 
    print result 

输出：

() 
('A1',) 
('A1', 'B1') 
('A1', 'B1', 'C1') 
('A1', 'B1', 'C2') 
('A1', 'B2') 
('A1', 'B2', 'C1') 
('A1', 'B2', 'C2') 
('A1', 'B3') 
('A1', 'B3', 'C1') 
('A1', 'B3', 'C2') 
('A1', 'C1') 
('A1', 'C2') 
('B1',) 
('B1', 'C1') 
('B1', 'C2') 
('B2',) 
('B2', 'C1') 
('B2', 'C2') 
('B3',) 
('B3', 'C1') 
('B3', 'C2') 
('C1',) 
('C2',)