python中的product函数(可对多个列表进行排列组合)

product函数用于求多个可迭代对象的笛卡尔积

举例:

使用product函数

from itertools import product

a = [1, 2, 3]
b = [4, 5, 6]
for n in product(a, b):
    print(n)
(1, 4)
(1, 5)
(1, 6)
(2, 4)
(2, 5)
(2, 6)
(3, 4)
(3, 5)
(3, 6)

用双循环遍历

a = [1, 2, 3]
b = [4, 5, 6]
for i in a:
    for j in b:
        print((i, j))
(1, 4)
(1, 5)
(1, 6)
(2, 4)
(2, 5)
(2, 6)
(3, 4)
(3, 5)
(3, 6)

这个例子中只有a和b两个列表,product(a, b)也可使用双循环遍历a和b后输出
如果求多个列表的笛卡尔积时,建议使用product函数

你可能感兴趣的:(python,python)