Элегантный способ получения комбинаций для списка итераций

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. Если ваши итерации не являются списками, вам нужно будет использовать 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',)