欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  后端开发

python 排列组合之itertools

程序员文章站 2022-04-18 07:57:32
...
python 2.6 引入了itertools模块,使得排列组合的实现非常简单:

代码如下:


import itertools


有序排列:e.g., 4个数内选2个排列:

代码如下:


>>> print list(itertools.permutations([1,2,3,4],2))
[(1, 2), (1, 3), (1, 4), (2, 1), (2, 3), (2, 4), (3, 1), (3, 2), (3, 4), (4, 1), (4, 2), (4, 3)]


无序组合:e.g.,4个数内选2个:

代码如下:


>>> print list(itertools.combinations([1,2,3,4],2))
[(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]