python 进阶技巧

1. itertools

作用：遍历一个集合中元素的所有可能的排列或组合

排列

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):  #3个元素的组合
    print(c)
for c in combinations(items,2):
    print(c) 
for c in combinations(items,1):  #1个元素的组合
    print(c)

('a', 'b', 'c')
('a', 'b')
('a', 'c')
('b', 'c')
('a',)
('b',)
('c',)

可重复组合

对combinations()来讲，元素顺序已经不重要了，就是说组合(‘a’,‘b’)跟(‘b’,‘a’)一样（最终只会输出其中一个）
在计算组合时，若元素已被选取就会从候选中剔除（若a已被选，接下来不会再考虑它）
itertools.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')

product

product(A,B)函数,返回A和B中的元素组成的笛卡尔积的元组

import itertools
for item in itertools.product([1,2,3,4],[100,200]):
    print(item)
    '''
(1, 100)
(1, 200)
(2, 100)
(2, 200)
(3, 100)
(3, 200)
(4, 100)
(4, 200)
    '''

product(A, B) 和 ((x,y) for x in A for y in B)一样.
它的一般使用形式如下：

itertools.product(*iterables, repeat=1)
iterables是可迭代对象,repeat指定iterable重复几次,即:

product(A,repeat=3)等价于product(A,A,A)