2016-11-18 119 views
1

我有以下列表:找到2 ^元素的n -2个组合在一个列表

list1 = ['g1','g2','g3','g4'] 

我想找到2^n-2组合,其中n是列表中的项目的总数。对于n = 4,结果应该是2^4 -2 = 14,即14个组合。

的组合方式如下:

[[['g1'],['g2','g3','g4']],[['g2'],['g1','g3','g4']], [['g3'],['g1','g2','g4']],['g4'],['g1','g2','g3']],[['g1','g2'],['g3','g4']],[['g1','g3'],['g2','g4']],[['g1','g4'],['g3','g4']],[['g2','g3'],['g1','g4']], 
[['g2','g4'],['g1','g3']],[['g3','g4'],['g1','g2']],[['g1','g2','g3'],['g4']],[['g2','g3','g4'],['g1']],[['g3','g4','g1'],['g2']],[['g4','g1','g2'],['g3']]] 

我知道一种方法: 在第一次迭代中采取单个元件,并把它放入一个列表和第二列表的其它元素:['g1'],['g2','g3','g4'] 在第二次迭代中采取在一个2种元素列表和第二列表中的其他元素。 ['g1','g2'],['g1','g4'] 有没有其他方法? 我正在用python编写这个程序。 我的方法是昂贵的。有没有任何库方法可以快速执行此操作。

+1

你是什么意思的代价?无论实现如何,生成组合的算法复杂度都是指数级的。 – hyades

+0

我更新了我的答案,以更好地反映您的要求。 –

回答

3

下面是使用itertools

import itertools as iter 

list1 = ['g1', 'g2', 'g3', 'g4'] 
combinations = [iter.combinations(list1, n) for n in range(1, len(list1))] 
flat_combinations = iter.chain.from_iterable(combinations) 
result = map(lambda x: [list(x), list(set(list1) - set(x))], flat_combinations) 
# [[['g1'], ['g4', 'g3', 'g2']], [['g2'], ['g4', 'g3', 'g1']], [['g3'], ['g4', 'g2', 'g1']],... 
len(result) 
# 14 
+0

你如何得到元组的第二部分? – hyades

+0

我不完全确定这是从轻微混淆的问题需要。他在组合之后,这只是元组的第一部分。 –

+0

OP的例子是一个有2个部分的有序集合分区的列表。这只会生成每个分区的一半。您需要将每个子集/组合与其补码进行配对。 –

2

itertools.combinations(可迭代,R)

从输入可迭代元件的返回ř长度子序列的功能性的方法。 组合按字典顺序排列。所以,如果输入 可迭代排序,组合元组将按排序的 顺序生成。

from itertools import combinations 
list1 = ['g1','g2','g3','g4'] 
for n in range(1,len(list1)): 
    for i in combinations(list1,n): 
     print(set(i), set(list1) - set(i)) 

出来:

{'g1'} {'g2', 'g3', 'g4'} 
{'g2'} {'g1', 'g3', 'g4'} 
{'g3'} {'g1', 'g2', 'g4'} 
{'g4'} {'g1', 'g2', 'g3'} 
{'g1', 'g2'} {'g3', 'g4'} 
{'g1', 'g3'} {'g2', 'g4'} 
{'g1', 'g4'} {'g2', 'g3'} 
{'g2', 'g3'} {'g1', 'g4'} 
{'g2', 'g4'} {'g1', 'g3'} 
{'g3', 'g4'} {'g1', 'g2'} 
{'g1', 'g2', 'g3'} {'g4'} 
{'g1', 'g2', 'g4'} {'g3'} 
{'g1', 'g3', 'g4'} {'g2'} 
{'g2', 'g3', 'g4'} {'g1'} 

你可以试试这个

+0

这与我的回答非常相似,并没有考虑到OP希望所有长度为1的组合都到'len(list1)'。 –

+0

只是在它上面,这很简单 –

+0

看到我的问题的操作。在这里,我不需要所有的nC2元素,我需要的是第一个元素在一个列表或元组中,第二个列表或元组中的其他元素。 ('g1','g2'),('g3','g4')等等('g1'),('g2','g3','g4')等 –

0

我想从中国编码器(我猜)的解决方案。这是我自己的解决方案:

import itertools 


def flatten(*z): 
    return z 


list1 = ['g1','g2','g3','g4'] 

sublists = [] 
for i in range(1, len(list1)): 
    sublists.extend(itertools.combinations(list1, i)) 

pairs = [] 
for a, b in itertools.product(sublists, sublists): 
    if len(a) + len(b) == len(list1) and \ 
    len(set(flatten(*a, *b))) == len(list1): 
     pairs.append((a, b)) 

print(pairs, len(pairs)) 
+1

既然你把powerset(减去空集和整集)作为一个列表来实现,那么你可以用它自己的相反的方式将其拉链:比如'pairs = list(zip(sublists,reversed(sublists)))'而不是使用'itertools.product'和过滤。 –

+0

这很酷使用反转和拉链! –

相关问题