itertools模块中的product方法
itertools模块中的product方法
itertools.product(*iterables[, repeat])
笛卡尔积
创建一个迭代器，生成表示item1，item2等中的项目的笛卡尔积的元组，repeat是一个关键字参数，指定重复生成序列的次数。
代码如下:
1 def product(*args, **kwds): 2 # product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy 3 # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111 4 pools = map(tuple, args) * kwds.get('repeat', 1) 5 result = [[]] 6 for pool in pools: 7 result = [x+[y] for x in result for y in pool] 8 for prod in result: 9 yield tuple(prod)
例子
代码如下:
1 import itertools 2 a = (1, 2, 3) 3 b = ('A', 'B', 'C') 4 c = itertools.product(a,b) 5 for elem in c: 6 print(elem) 7 (1, 'A') 8 (1, 'B') 9 (1, 'C') 10 (2, 'A') 11 (2, 'B') 12 (2, 'C') 13 (3, 'A') 14 (3, 'B') 15 (3, 'C')
