[Level 1 / Python] 행렬의 덧셈

문제 설명

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

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

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

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

풀이 방법

행렬 연산이 가능한 파이썬 라이브러리인 numpy를 이용하여 풀었습니다.

소스 코드

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

다른 사람의 풀이

def sumMatrix(A,B):
    answer = [[c + d for c, d in zip(a, b)] for a, b in zip(A,B)]
    return answer