Как сложить целые числа в две матрицы с помощью функции zip () в Python - PullRequest
1 голос
/ 05 апреля 2019

Я не могу понять, как сложить целые числа в две матрицы, используя функцию zip().Вот что у меня есть:

matrix_a = [[3,6],[4,5]]

matrix_b = [[5,8],[6,7]]

Мне нужно распечатать (используя zip()):

[[8,14],[10,12]]

1 Ответ

1 голос
/ 05 апреля 2019

Следующее понимание списка поможет:

>>> [[x + y for x, y in zip(a, b)] for a, b in zip(matrix_a, matrix_b)]
[[8, 14], [10, 2]]

Если вы хотите версию, которая использует циклы:

result = []
for a, b in zip(matrix_a, matrix_b):
    current_list = []
    for x, y in zip(a, b):
        current_list.append(x + y)
    result.append(current_list)
>>> result
[[8, 14], [10, 2]]

Однако, мне определенно нравится больше версия понимания, этолегче читать.

...