행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.

제한 조건
행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.
입출력 예
arr1 arr2 return
[[1,2],[2,3]] [[3,4],[5,6]] [[4,6],[7,9]]
[[1],[2]] [[3],[4]] [[4],[6]]

풀이

def solution(arr1, arr2):
    for i in range(len(arr1)): # 세로 길이만큼 (행 개수만큼)
        for j in range(len(arr1[0])): # 가로 길이만큼 (열 개수만큼)
            arr1[i][j] += arr2[i][j]  
    return arr1
            

결과

남의 코드

import numpy as np
def sumMatrix(A,B):
    A=np.array(A)
    B=np.array(B)
    answer=A+B
    return answer.tolist()

numpy를 사용해서 쉽게 행렬의 덧셈을 구할 수 있다.

참고

https://excelsior-cjh.tistory.com/25 : numpy코드 참고