最近写代码时遇到排列组合问题,发现python中的itertools库用起来比较方便。itertools库中的permutations函数可以输出可迭代对象的全排列情况,而combinations函数可以输出可迭代对象的全组合情况。
一、语法说明
import itertools
result=itertools.permutations(iterable,r)
其中result为对迭代对象处理之后返回的结果,数据类型为'itertools.permutations',如果需要的话,可以通过list()转化为列表。转化为列表之后的元素的数据类型为元组。元素默认排列顺序为迭代对象字典序上的从小到大;
iterable为需要排列的迭代对象,包括列表、字符串、元组、字典(只对键进行全排列);
r为单个排列元素的长度,不修改的话,默认为迭代对象的元素个数。
二、具体实施
全组合函数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)