如何使用組合

Question

我有6名名單，也就是說，

a=[1,1,0,0] 
b=[0,1,1,0] 
c=[0,0,1,1] 
d .... until f. 

我要生成總和的結果從2所列出開始，直到6所列出清單的所有可能的組合。例如，我想計算a + b，a + c，.. a + f的結果。然後，a + b + c，a + b + d ......等。我知道鋤頭計算兩個或三個列表的結果，但我堅持如何生成列表的組合。我試圖定義列表的列表，然後使用組合，帶參數2生成所有可能的組合，2 3名列表（爲例）如下：

import itertools 

alphabet = [[0,0,0],[0,0,1],[0,1,0]] 

combos = itertools.combinations(alphabet, 2) 

usable_combos = [] 
for e in combos: 
    usable_combos.append(e) 

但這根本不會產生任何東西。當我打印usable_combos，我得到：

[[0,0,0],[0,0,1],[0,1,0]] 

我的問題是：使用的組合，我怎麼能生產的6套不同的我有所有可能的組合（2至6種組合）？

Answer 1

使用range(1, len(lis)+1)可獲得傳遞給combinations的第二個參數（r）的值。或者range(2, len(lis)+1)如果你想開始從2

>>> from itertools import combinations 
>>> lis = [[0,0,0],[0,0,1],[0,1,0]] 
>>> for i in range(1, len(lis)+1): 
...  for c in combinations(lis,i): 
...   print c 
...   
([0, 0, 0],) 
([0, 0, 1],) 
([0, 1, 0],) 
([0, 0, 0], [0, 0, 1]) 
([0, 0, 0], [0, 1, 0]) 
([0, 0, 1], [0, 1, 0]) 
([0, 0, 0], [0, 0, 1], [0, 1, 0]) 

正如指出的評論可能@abarnert，可能是你想要的：

>>> from pprint import pprint 
>>> from itertools import chain 
>>> flatten = chain.from_iterable 
>>> ans = [list(flatten(c)) for i in range(2, len(lis)+1) for c in permutations(lis,i)] 
>>> pprint(ans) 
[[0, 0, 0, 0, 0, 1], 
[0, 0, 0, 0, 1, 0], 
[0, 0, 1, 0, 0, 0], 
[0, 0, 1, 0, 1, 0], 
[0, 1, 0, 0, 0, 0], 
[0, 1, 0, 0, 0, 1], 
[0, 0, 0, 0, 0, 1, 0, 1, 0], 
[0, 0, 0, 0, 1, 0, 0, 0, 1], 
[0, 0, 1, 0, 0, 0, 0, 1, 0], 
[0, 0, 1, 0, 1, 0, 0, 0, 0], 
[0, 1, 0, 0, 0, 0, 0, 0, 1], 
[0, 1, 0, 0, 0, 1, 0, 0, 0]]