在领扣刷题的过程中遇到需要使用笛卡尔积的解法,因为使用的是python3版本解决,所以在python版本中提供了一个库函数可以模拟笛卡尔乘积的元素组成过程:itertools.product,一般来说比较常使用的是在product方法中传递进A与B两个可迭代的对象进行组合,官方提供的解释如下:https://docs.python.org/zh-cn/3.8/library/itertools.html#itertools.product
# product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
上面的例子很好地解释了这个函数的常见用法,其实本质上来说是两个for循环的嵌套
依次取出A与B中的每一个元素组成元组返回,代码如下:
import itertools
if __name__ == '__main__':
l1 = [1, 2, 3]
l2 = ["a", "b", "c"]
for cur in itertools.product(l1, l2):
print(cur, end=" ")
(1, 'a') (1, 'b') (1, 'c') (2, 'a') (2, 'b') (2, 'c') (3, 'a') (3, 'b') (3, 'c')