[Programmers] 행렬의 덧셈

행렬의 덧셈

level 1

연습문제

 

python3

def solution(a, b):
    return [[a[m][n]+b[m][n] for n in range(len(a[0]))] for m in range(len(a))]

list comprehension을 이용.

두 행렬의 크기가 같기 때문에 가능한 방법.

 

import numpy as np

def sumMatrix(A,B):
    A = np.array(A)
    B = np.array(B)
    answer = A+B
    return answer.tolist()

혹은 numpy를 이용해도 된다.