首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >具有`itertools.combinations`的子集和

具有`itertools.combinations`的子集和
EN

Stack Overflow用户
提问于 2014-01-29 09:36:43
回答 1查看 2.1K关注 0票数 4

我在python中有一个整数列表,假设:

代码语言:javascript
复制
weight = [7, 5, 3, 2, 9, 1]

我应该如何使用itertools.combinations来查找这些整数的所有可能的sums子集。因此,我得到了(3个整数的期望输出示例- weight = 7,5,3

代码语言:javascript
复制
sums = [ [7], [7+5], [7+3], [7+5+3], [5], [5+3], [3] ]

与这些重量相关联,我有另一个名为Associated的数组,它是一个列表列表,其中包含行李名称及其对应的重量,格式如下:

代码语言:javascript
复制
luggages = [["samsonite", 7], ["Berkin", 5], ["Catelli", 3] .....]

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

代码语言:javascript
复制
weight = numpy.array([c[1] for c in luggages]) 

我可以对需要的行李名称执行此操作。我尝试以这种方式使用itertools.combinations (根据建议):

代码语言:javascript
复制
comb = [combinations(weight, i) for i in range(len(luggages))]

我的目标:打印出所有可能的行李名称子集,我可以在旅行中携带,给定满足子集和恰好等于23 KG的条件的所有子集的组合的max_weight = 23 kg。简而言之,我必须打印出一个带有行李箱名称的列表,如果将其重量相加,则恰好等于max_weight = 23。记住:行李只能在每个子集中选择一次,但它们可以出现在尽可能多的子集中。此外,每个子集中的项目数量也是无关紧要的:可以是1个行李,2个,3个……只要它们的和恰好等于23。

EN

回答 1

Stack Overflow用户

发布于 2014-01-29 10:09:27

在调查旅行推销员,是不是?您可以使用每个人都喜欢的Python特性list comprehensions来完成此操作:

代码语言:javascript
复制
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)
票数 5
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/21420452

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档