itertools模块中的product方法

itertools模块中的product方法

itertools.product(*iterables[, repeat])

笛卡尔积

创建一个迭代器，生成表示item1，item2等中的项目的笛卡尔积的元组，repeat是一个关键字参数，指定重复生成序列的次数。

代码如下:

 def product(*args, **kwds):
     # product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
     # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
     pools = map(tuple, args) * kwds.get('repeat', 1)
     result = [[]]
     for pool in pools:
         result = [x+[y] for x in result for y in pool]
     for prod in result:
         yield tuple(prod)

例子

代码如下:

 import itertools
 a = (1, 2, 3)
 b = ('A', 'B', 'C')
 c = itertools.product(a,b)
 for elem in c:
     print(elem)
 (1, 'A')
 (1, 'B')
 (1, 'C')
 (2, 'A')
 (2, 'B')
 (2, 'C')
 (3, 'A')
 (3, 'B')
 (3, 'C')