[프로그래머스 코딩테스트] lv1. 행렬의 덧셈

문제 설명

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

나의 풀이

import numpy as np

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

이 문제는 numpy를 사용하면 매우 쉽게 풀 수 있습니다. 넘파이는 행렬, 텐서를 이용할 때 사용되는 패키지로 데이터 사이언스 쪽을 공부하게 된다면 반드시 알아야 하는 패키지입니다. 각 행렬을 numpy array로 만들어 더해주면 쉽게 값을 구할 수 있습니다.

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/12950

코딩테스트 연습 - 행렬의 덧셈