[Programmers] 프로그래머스 문제풀이 행렬의 덧셈

한 번 스스로 풀어보는것을 권장합니다.

문제는 여기서 풀어볼 수 있습니다.

행렬의 덧셈

문제 설명

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

제한 조건

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

입출력 예

arr1	arr2	return
[[1,2],[2,3]]	[[3,4],[5,6]]	[[4,6],[7,9]]
[[1],[2]]	[[3],[4]]	[[4],[6]]

풀이

프로그래머스에서 numpy 모듈을 사용할 수 있어 굉장히 쉽게 풀 수 있지만, 모듈을 사용하지 않고 풀어보기를 권장한다.

import numpy as np


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