product 用于求多个可迭代对象的笛卡尔积(Cartesian Product),它跟嵌套的 for 循环等价.即:
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)
from itertools import product
A = [1, 2, 3]
B = [100, 200, 300]
for item in product(A, B):
print(item)
# 输出
(1, 100)
(1, 200)
(1, 300)
(2, 100)
(2, 200)
(2, 300)
(3, 100)
(3, 200)
(3, 300)
from itertools import product
d ={'1':['a','b'], '2':['c','d','e'],'3':['f','h']}
for combo in product(*[d[k] for k in sorted(d.keys())]):
print(combo)
#输出
('a', 'c', 'f')
('a', 'c', 'h')
('a', 'd', 'f')
('a', 'd', 'h')
('a', 'e', 'f')
('a', 'e', 'h')
('b', 'c', 'f')
('b', 'c', 'h')
('b', 'd', 'f')
('b', 'd', 'h')
('b', 'e', 'f')
('b', 'e', 'h')