행렬의 덧셈

해당 문제는 위와 같이 행렬을 받아 그 행렬 각 요소에 대응하는 값을 더한 행렬을 리턴하는 문제였다.

아래는 내가 푼 두 가지 방법이다.

def solution(a, b):
    answer=a
    for i in range(len(a)):
        for k in range(len(a[i])):
            answer[i][k]= a[i][k] + b[i][k]
    
    
     return answer

import numpy as np
def solution(a, b):
    a=np.array(a)
    b=np.array(b)
    result=a+b

    return result.tolist()

---

이것은 zip을 사용한 방법. 아직 내 갈길은 멀고, 코드는 아름답구나...

def solution(a,b):
    return [[a+b for a,b in zip(x,y)] for x,y in zip(a,b)]