如何在 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))
这是我的代码的一部分:
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))