python 实现排列组合

Posted 2021-03-25 Skyline

对于一个数组（或任何可以迭代的元素集），可以通过itertools包中的permutations和combinations轻松完成排列，组合

python3中permutations和combinations返回的是一个迭代器，可以通过list转化为一个列表，方便我们进一步处理

具体用法看下面的例子

import itertools
from itertools import permutations, combinations

print("Permutations of ‘bar‘")
print(list(permutations(‘bar‘)))

#输出结果
#Permutations of ‘bar‘
#[(‘b‘, ‘a‘, ‘r‘), (‘b‘, ‘r‘, ‘a‘), (‘a‘, ‘b‘, ‘r‘), (‘a‘, ‘r‘, ‘b‘), (‘r‘, ‘b‘, ‘a‘), (‘r‘, ‘a‘, ‘b‘)]

print("Combinations of 2 letters from ‘bar‘")
print(list(combinations(‘bar‘, 2)))

#输出结果
#Combinations of 2 letters from ‘bar‘
#[(‘b‘, ‘a‘), (‘b‘, ‘r‘), (‘a‘, ‘r‘)]