프로그래머스) 행렬의 덧셈. 파이썬

numpy를 이용하여, 파이썬 문제를 처음 풀어봤음.

for loop가 없어 굉장히 군더더기 없이 간단하고 깔끔하다.

 

주의할 점은, numpy는 ndarray이고, 답안 변수 answer는 list형식이라, tolist()를 써서 형변환을 해줘야 한다는 것. 

 

import numpy as np

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