我在python中有一个整数列表,假设:
weight = [7, 5, 3, 2, 9, 1]我应该如何使用itertools.combinations来查找这些整数的所有可能的sums子集。因此,我得到了(3个整数的期望输出示例- weight = 7,5,3
sums = [ [7], [7+5], [7+3], [7+5+3], [5], [5+3], [3] ]与这些重量相关联,我有另一个名为Associated的数组,它是一个列表列表,其中包含行李名称及其对应的重量,格式如下:
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。
发布于 2014-01-29 10:09:27
在调查旅行推销员,是不是?您可以使用每个人都喜欢的Python特性list comprehensions来完成此操作:
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)https://stackoverflow.com/questions/21420452
复制相似问题