首页 > 后端开发 > Python教程 > 如何使用 Python 的'itertools.combinations”生成集合的所有子集?

如何使用 Python 的'itertools.combinations”生成集合的所有子集?

Linda Hamilton
发布: 2024-12-11 13:52:12
原创
370 人浏览过

How to Generate All Subsets of a Set Using Python's `itertools.combinations`?

如何使用 itertools.combinations 生成集合的所有子集

在 Python 中,itertools.combinations 模块提供了一种简单高效的方法用于生成集合的幂集。具体操作方法如下:

1

2

3

4

5

6

from itertools import chain, combinations

 

def powerset(iterable):

    "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"

    s = list(iterable)

    return chain.from_iterable(combinations(s, r) for r in range(len(s)+1))

登录后复制

例如,要查找集合 {0, 1, 2, 3} 的所有子集,您可以使用以下代码:

1

2

>>> list(powerset([0, 1, 2, 3]))

[(), (0,), (1,), (2,), (3,), (0, 1), (0, 2), (0, 3), (1, 2), (1, 3), (2, 3), (0, 1, 2), (0, 1, 3), (0, 2, 3), (1, 2, 3), (0, 1, 2, 3)]

登录后复制

请注意,空元组 () 包含在幂集中,因为它代表空子集。

如果您不想拥有结果中的空元组,您可以修改组合循环中的范围,如下所示:

1

2

3

4

def powerset(iterable):

    "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"

    s = list(iterable)

    return chain.from_iterable(combinations(s, r) for r in range(1, len(s)+1))

登录后复制

这将从返回的子集中排除空元组。

以上是如何使用 Python 的'itertools.combinations”生成集合的所有子集?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板