[코딩테스트 연습] 행렬의 덧셈

문제

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

 

❗제한사항❗

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

 

입출력 예

✔풀이

import numpy as np

def solution(arr1, arr2):

    arr_1=np.array(arr1)
    arr_2=np.array(arr2)
    sum_arr=arr_1 + arr_2
    answer = sum_arr.tolist()
    
   
    return answer

📝설명

문제에서는 행렬간의 덧셈의 결과를 도출하라고 하고 있다. 

우선 행렬을 계산하기 위해서, numpy를 호출하여 np로 별칭을 붙였다.

그 후 각 행렬에 np.array()를 이용하여 행렬화하였고, sum()을 활용하여 행렬간의 덧셈을 해결했다.

return해야 하는 값은 리스트 형태이기 때문에, tolist()로 리스트화하여 answer에 저장했다.

 

👑모범 답안

🔎괴물들...

 

👀느낀 점

: 모범 답안에 비하면 길지만, 배운 numpy로 문제를 푸니 뿌듯합니다... 

 

💻 문제 출처

코딩테스트 연습 - 행렬의 덧셈 | 프로그래머스 스쿨 (programmers.co.kr)