python的几种有用的函数 itertools库

import itertools

b = [('a', -1), ('b', 'b1'), ('c', 'cs')]

#笛卡尔积
list(itertools.product(*b))

[('a', 'b', 'c'),
 ('a', 'b', 'cs'),
 ('a', 'b1', 'c'),
 ('a', 'b1', 'cs'),
 (-1, 'b', 'c'),
 (-1, 'b', 'cs'),
 (-1, 'b1', 'c'),
 (-1, 'b1', 'cs')]


#排列
list(itertools.permutations(b, 2))

[(('a', -1), ('b', 'b1')),
 (('a', -1), ('c', 'cs')),
 (('b', 'b1'), ('a', -1)),
 (('b', 'b1'), ('c', 'cs')),
 (('c', 'cs'), ('a', -1)),
 (('c', 'cs'), ('b', 'b1'))]


#组合,没有重复
list(itertools.combinations(b, 2))
[(('a', -1), ('b', 'b1')),
 (('a', -1), ('c', 'cs')),
 (('b', 'b1'), ('c', 'cs'))]


#组合,有重复
list(itertools.combinations_with_replacement(b, 2))


[(('a', -1), ('a', -1)),
 (('a', -1), ('b', 'b1')),
 (('a', -1), ('c', 'cs')),
 (('b', 'b1'), ('b', 'b1')),
 (('b', 'b1'), ('c', 'cs')),
 (('c', 'cs'), ('c', 'cs'))]

 

转载于:https://my.oschina.net/whx403/blog/829190