Python - 如何从嵌套列表中获取所有可能的组合

Question

我有一个像这样的嵌套列表：values = [['DNO', 0.2], ['Equinor', 0.4], ['Petoro', 0.2], ['Total', 0.2]]如何获得总和（每个子列表的第二个元素）大于 0.5 的所有可能的元素组合？这就是我正在使用的：def getCombinations(values, min_len):    combo = "\n"    numbers = []    doc = {}    for val in values:        doc[val[0]] = val[1]        numbers.append(val[1])    result = [seq for i in range(len(numbers), 0, -1) for seq in itertools.combinations(numbers, i) if sum(seq) >= 0.5]    temp = doc.copy()    for r in result:        doc = temp.copy()        if len(r) >= min_len:            for rr in r:                combo = combo + get_key(doc, rr) + " "                doc.pop(get_key(doc, rr))            combo = combo + "\n"    return combo当上面列表中有多个值（例如 0.2）时，我的算法会出现一些问题。目前它返回的是min_length=3：Total Equinor Petoro DNOTotal Equinor Petoro Total Equinor Petoro Total Petoro DNO Equinor Total Petoro 

Answer 1

您可以使用这样的列表理解：

解释：

• 第一个for定义了组合的长度。values使用从 2 到 的长度的每个长度。

• 第二个for创建实际组合

• 使用if生成器方法对项目数进行求和

from itertools import combinations

combis = [

    item

    for length in range(2, len(values)+1)

    for item in combinations(values, length)

    if sum(i[1] for i in item) >= 0.5

]

Answer 2

以下应该有效：

import itertools  

result=[] 

for k in range(2,len(values)+1):

    temp=[tuple(x[0] for x in i) for i in list(itertools.combinations(values,k))if sum([p[1] for p in i]) >0.5]

    result.append(temp)

result=sum(result, [])

print(result)

输出：

[('DNO', 'Equinor'), ('Equinor', 'Petoro'), ('Equinor', 'Total'), ('DNO', 'Equinor', 'Petoro'), ('DNO', 'Equinor', 'Total'), ('DNO', 'Petoro', 'Total'), ('Equinor', 'Petoro', 'Total'), ('DNO', 'Equinor', 'Petoro', 'Total')]