python - 如何使用 Python 中的 zip() 函数将两个矩阵中的整数相加

Question

我无法弄清楚如何使用 zip() 函数将两个矩阵中的整数相加。这是我所拥有的:

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

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

我需要打印出来(使用zip()):

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

Answer 1

下面的列表理解就可以解决问题:

>>> [[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]]

但是，我绝对更喜欢理解版本，因为它更容易阅读。