Как использовать itertools.product () в python до l oop для нескольких массивов? - PullRequest
0 голосов
/ 27 марта 2020

Это часть моего кода:

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 "" "Количество массивов" ""

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)].

1 Ответ

0 голосов
/ 27 марта 2020

Как то так?

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

list(product(*arrays))
...