查找組合列表

Question

我正在使用python 2.7。給定一組列表，是否有快速有效的方法來生成組合？輸入設置總是在每一組輸出集合中的一個項目都有2

長度：

[set(['item1']), set(['item2']), set(['item3'])] 

要：

[set(['item1','item2']), set(['item2','item3']), set(['item3','item1'])] 

Answer 1

由於您的集列表中的元素所有的1元素集合，你正在尋找的組合只是這些集合的2元素子集合。您可以獲取他們是這樣的：

>>> import itertools 

>>> sets = [set(['item1']), set(['item2']), set(['item3'])] 
>>> elements = set() 
>>> for s in sets: elements.update(s) 

從而

>>> elements 
{'item1', 'item2', 'item3'} 

接着，就這樣：

>>> pairs = [set(combo) for combo in itertools.combinations(elements,2)] 
>>> pairs 
[{'item1', 'item2'}, {'item1', 'item3'}, {'item2', 'item3'}] 

Answer 2

正如約翰指出itertools可能會有所幫助。這裏有一個簡單的例子：

import itertools as it 
sets = [set(range(0, 3)), set(range(2, 5)), set(range(4, 7))] 
comb = list(it.combinations(sets, r=2)) 
comb 

輸出：[({0, 1, 2}, {2, 3, 4}), ({0, 1, 2}, {4, 5, 6}), ({2, 3, 4}, {4, 5, 6})]

然後創建在每個迭代的交集：

comb_sets = [a.intersection(b) for a, b in comb] 
comb_sets 

輸出：[{2}, set(), {4}]