python - * 参数的 Itertools

Question

我想要一个函数来生成任意数量的数组的叉积。

# Code to generate cross product of 3 arrays
M = [1, 1]
N = [2, 3]
K = [4, 5]
for M, N, K in itertools.product(M, N, K)

如果我想使用 * 引入一个函数，有什么好的方法可以实现这一点？

我尝试了以下代码，但最终出现错误:“TypeError:'builtin_function_or_method'对象不可迭代”

# def cross_product(*inputs):
      return itertools.product(inputs)

cross_product(M, N, K)

Answer 1

实际上，您可以只使用解包而不使用辅助函数:

import itertools

L = [[1, 1],
     [2, 3],
     [4, 5]]

for x in itertools.product(*L):
    print(x)