[Py_Lv1] 행렬의 덧셈

문제 설명

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

제한 사항

• 행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

입출력 예

접근법

우선, 이 문제를 봤을때 최근 공부하고 있는 numpy 라이브러리가 떠올랐다.

머리속에 자세하게 그려지지 않아 for문으로 문제를 풀어보았다.

그리고 numpy 함수를 이용해 다시 구현해보았다.

확실히 파이썬을 공부한다면 2번째 해결방법이 더 어울리는것 같다.

라이브러리에 대단함을 다시 한번 느낄 수 있는 문제였다.

어렵지 않은 문제라 별도의 설명없이 마무리하겠다.

=============================================================

해결방법 1

def solution(arr1, arr2):
    for i in range(len(arr1)):
        for j in range(len(arr1[i])):
             arr1[i][j] += arr2[i][j]
    return arr1

해결방법 2

import numpy as np

def solution(arr1, arr2):
    arr_1 = np.array(arr1)
    arr_2 = np.array(arr2)
    answer = (arr_1 + arr_2)
    return answer.tolist()