如何在 python 中使用 itertools.product() 来遍历多个数组?

How do I use itertools.product() in python to loop over multiple arrays?

这是我的代码的一部分:

m,n=list(map(int,input().split()))
for i in range(m):
    L=list(map(int,input().split()))
    if(i==0):
        K=L.copy()
        continue
    K=list(product(list(K),list(L)))

程序的输入是:

4“”"No.of arrays"”

2 5 3

3 7 4

1 5 3

3 5 1

我已经使用 itertools.product 在 4 个数组之间执行笛卡尔积。

我得到的输出是这样的形式: [(((2, 3), 5),1).....(((3,4),3),5)]。 我怎样才能改进我的代码,以便我得到以下形式的输出: [(2,3,5,1)......(3,4,3,5)]。

是这样的吗?

arrays = [[2, 5, 3], [3, 7, 4], [1, 2, 3], [3, 5, 1]]

list(product(*arrays))