python中的笛卡尔乘积

大家好，欢迎来到IT知识分享网。

在领扣刷题的过程中遇到需要使用笛卡尔积的解法，因为使用的是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’)