带有“itertools.combinations”的子集和

Question

我在 python 中有一个整数列表，比如说：

weight = [7, 5, 3, 2, 9, 1]

我应该如何使用 itertools.combinations 来找到这些整数的所有可能的总和子集。 这样我就得到了（一个带有 3 个整数的期望输出示例 - weight = [7, 5, 3]：

sums = [ [7], [7+5], [7+3], [7+5+3], [5], [5+3], [3] ]

与这些重量相关，我有另一个名为“行李箱”的数组，它是一个包含行李名称及其对应重量的列表列表，格式如下：

luggages = [["samsonite", 7], ["Berkin", 5], ["Catelli", 3] .....]

我以这种方式创建了一个名为 weight 的数组。

weight = numpy.array([c[1] for c in luggages]) 

我可以为需要的行李名称执行此操作。 我尝试以这种方式使用 itertools.combinations（根据建议）：

comb = [combinations(weight, i) for i in range(len(luggages))]

我的目标：在满足子集总和等于 23 KG 的条件的每个子集的所有组合的 max_weight = 23 kg 的情况下，打印出我可以在旅行中携带的所有可能的行李名称子集。 用更简单的术语来说，我必须打印出一个包含行李名称的列表，如果它的重量相加将等于max_weight = 23 完全正确。 请记住：行李只能在每个子集中选择一次，但它们可以出现在尽可能多的子集中。此外，每个子集中的项目数是无关紧要的：它可以是 1 件行李、2、3 件……只要它们的总和正好等于 23。

Answer 1

为旅行推销员工作，是吗？您可以使用每个人都喜欢的 Python 功能列表推导来做到这一点：

weight = [7, 5, 3, 2, 9, 1]

cmb = []
for x in range(1, len(weight) + 1):
    cmb += itertools.combinations(weight, x)

#cmb now contains all combos, filter out ones over the limit

limit = 23

valid_combos = [i for i in cmb if sum(i) == limit]

print(valid_combos)