Застрял с добавлением матрицы в 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