python迭代所有可能的组合或排列进行迭代

我们想对一系列元素所有可能的组合或排列进行迭代，为解决这个问题，itertools模块中提供了3个函数。第一个是itertools.permutations()，它接受一个元素集合，将其中所有的元素重排为所有可能的情况，并以元祖序列的形式返回。如下：

In [10]: items = ['a', 'b', 'c']

In [11]: from itertools import permutations

In [12]: 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')

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

In [13]: for p in permutations(items, 2):
    ...:     print(p)
    ...:
('a', 'b')
('a', 'c')
('b', 'a')
('b', 'c')
('c', 'a')
('c', 'b')

使用itertools.combinations()可产生输入序列中所有元素的全部组合形式。如下：

In [14]: from itertools import combinations

In [15]: for c in combinations(items, 3):
    ...:     print(c)
    ...:
('a', 'b', 'c')

In [16]: for c in combinations(items, 2):
    ...:     print(c)
    ...:
('a', 'b')
('a', 'c')
('b', 'c')

In [17]: for c in combinations(items, 1):
    ...:     print(c)
    ...:
('a',)
('b',)
('c',)

对于conbinations()来说，元素之间的实际顺序是不考虑的。当产生组合时，已经选择过的元素将从可能的候选元素中移除掉(即，如果‘a’已经选择过了，那么就将它从考虑范围中去掉)。itertools.combinations_with_replacement()函数解除了这一限制，允许相同的元素得到多次选择。如下：

In [22]: from itertools import combinations_with_replacement

In [23]: 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')