python-遍历所有可能的组合和排列

4.9 如果我们相对一系列元素所有可能的组合或排列进行遍历

• 可以使用itertools模块中提供了3个函数。第一个是itertools.permutations() 它接受一个元素集合，将其中所有的元素重排列为所有可能的情况，并以元组祖列的形式返回

>>> items = ['a','b','c']
>>> from itertools import permutations
>>> for p in permutations(items):
...     print(p)
...
('a', 'b', 'c')
('a', 'c', 'b')
('b', 'a', 'c')
('b', 'c', 'a')
('c', 'a', 'b')
('c', 'b', 'a')
>>>

• 如果想得到较短长度的所有排列，可以提供一个可选长度参数，如

>>> for p in permutations(items,2):
...     print(p)
...
('a', 'b')
('a', 'c')
('b', 'a')
('b', 'c')
('c', 'a')
('c', 'b')

• 使用itertools.combinations()可以产生输入序列中所有元素的组合。

>>> from itertools import combinations
>>> for c in combinations(items,3):
...     print(c)
...
('a', 'b', 'c')
>>> for c in combinations(items,2):
...     print(c)
...
('a', 'b')
('a', 'c')
('b', 'c')
>>>

• 对combinations来产生组合时，是将元素从候选元素中移除，这样保证每个元素只被选中一次

• combinations_with_replacement函数可以使元素被重复选择

>>> from itertools import combinations_with_replacement
>>> for c in combinations_with_replacement(items,3):
...     print(c)
...
('a', 'a', 'a')
('a', 'a', 'b')
('a', 'a', 'c')
('a', 'b', 'b')
('a', 'b', 'c')
('a', 'c', 'c')
('b', 'b', 'b')
('b', 'b', 'c')
('b', 'c', 'c')
('c', 'c', 'c')
>>>