[Programmers] 행렬의 덧셈 - 12950

numpy array로 변환하여 연산하면 단순 덧셈으로 행렬 간의 덧셈이 가능하다 ! 

 

Question

 

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

 

 

Point

 

1. numpy array 사용
2. 이중 for문 사용

 

 

Code 1 : numpy 사용

 

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

 

Code 2 : 이중 for문 사용

 

def solution(arr1, arr2):
    answer = []
    
    for i,j in zip(arr1, arr2):
        sum = []
        for n,m in zip(i,j):
            sum.append(n+m)
        answer.append(sum)
    
    return answer

 

now me

On my github