Застрял с добавлением матрицы в Python

Пожалуйста, посмотрите на мой код для добавления матрицы в python и помогите мне решить проблему.

Код:

def matrix_addition(a, b):
# your code here
res = []  
for i in range(len(a)):
    for j in range(len(b)):
        sum = a[i][j] + b[i][j]
        res.append(sum)
return res

matrix_addition( [ [1, 2],[1, 2] ], [ [2, 3],[2, 3] ] )

Ожидаемый результат: [[3, 5], [3, 5]]

Мой вывод: [3, 5, 3, 5]

Как инициализировать вложенный список и включить в него некоторые переменные?

PS: я новичок в Python, поэтому ожидаю более легкого решения:)

1 ответ

Решение

Для начинающих в Python, обратите особое внимание на отступ, потому что это основа синтаксиса Python, а не конечный разделитель, как в большинстве языков / скриптов.

Вы не создаете массив для суммы и не добавляете его в правый цикл. Попробуй это:

def matrix_addition(a, b):
# your code here
res = []  
for i in range(len(a)):
    sum = []
    for j in range(len(b)):
        sum.append([i][j] + b[i][j])
    res.append(sum)
return res
Другие вопросы по тегам