变化无重复

Question

我一直在寻找一种有效的方法来检查列表 n 中 k 个元素的 sum 是否等于 x 强>值。我认为也许在numpy和intertools库中会有所帮助。

示例：

list = [1, 0, 1, 3, 4, 2, 2]
temp = list.copy()
x = 10
result = []
import random

while sum(result) != x:
    try:
        a = random.choice(temp)
        result.append(a)
        temp.remove(a)
    except IndexError:
        temp = list.copy()
        result = []
        if sum(result) > 10:
            result = []

对于大列表的随机选择这不是很有效。

Answer 1

我认为itertools.combinations是您所需要的

from itertools import combinations

l = [1, 0, 1, 3, 4, 2, 2]
x = 10

for i in range(len(l)):
    for c in combinations(l, i):
        if sum(c) == x:
            print(c)