求多个可迭代对象的笛卡尔积:itertools之product函数

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')
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值