[프로그래머스] 행렬의 덧셈 (Python)

문제 링크 (Level 1)

https://school.programmers.co.kr/learn/courses/30/lessons/12950?language=python3 

프로그래머스

 

나의 코드 1 (Python)

def solution(arr1, arr2):
    answer = []
    for i in range(len(arr1)): # 배열 전체 길이
        tmp = []
        for j in range(len(arr1[0])): # 배열 한 행의 길이
            tmp.append(arr1[i][j] + arr2[i][j]) # (i, j)
        answer.append(tmp)
    return answer

행렬을 하나씩 추출하여 덧셈 후 tmp에 행렬 형태로 만들어서 answer에 저장하는 방법

 

나의 코드 2 (Python) - numpy 사용

import numpy as np
def solution(arr1, arr2):
    a = np.array(arr1)
    b = np.array(arr2)
    answer = a + b
    return answer.tolist()

연산에 특화되어 있는 numpy 라이브러리를 사용하여 쉽게 계산하는 방법