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

코딩테스트 연습 > 연습문제 > 행렬의 덧셈


Numpy 모듈

numpy가 제공하는 numpy.array를 통해 행렬 계산


1차

import numpy as np

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

 

관련 이전글

[Python] TypeError: Object of type ndarray is not JSON serializable


2차

import numpy as np

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

    return answer

728x90