python中全组合函数(combinations)与全排列函数(permutations)

最近写代码时遇到排列组合问题,发现python中的itertools库用起来比较方便。itertools库中的permutations函数可以输出可迭代对象的全排列情况,而combinations函数可以输出可迭代对象的全组合情况。

一、语法说明

import itertools
result=itertools.permutations(iterable,r)

其中result为对迭代对象处理之后返回的结果,数据类型为'itertools.permutations',如果需要的话,可以通过list()转化为列表。转化为列表之后的元素的数据类型为元组。元素默认排列顺序为迭代对象字典序上的从小到大;

iterable为需要排列的迭代对象,包括列表、字符串、元组、字典(只对键进行全排列);

r为单个排列元素的长度,不修改的话,默认为迭代对象的元素个数。

二、具体实施

  1. 全组合函数combinations()

import itertools
a = [1, 2, 3]
b = itertools.combinations(a, 2)  # 含有两个元素的组合
for bi in b:  # 或 print(list(b))
    print(bi)

输出结果

(1, 2)
(1, 3)
(2, 3)

2. 全排列函数permutations()

c = itertools.permutations(a, 2)  # 全排列
for ci in c:  # 或 print(list(c))
    print(ci)

输出结果

(1, 2)
(1, 3)
(2, 1)
(2, 3)
(3, 1)
(3, 2)

3. 部分组合

a = [1, 2, 3]
for i in range(len(a)-1): 
    print((a[i],a[i+1]))
(1, 2)
(2, 3)

猜你喜欢

转载自blog.csdn.net/weixin_43734080/article/details/129356788
今日推荐