코딩테스트 연습 > 연습문제 > 행렬의 덧셈
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