[level 1] 행렬의 덧셈

문제

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

 

문제 풀이 접근법1

참고: https://yjs-program.tistory.com/324

[python] list 요소 2개씩 묶기

def solution(arr1, arr2):
        
    answer = [[i + j for i, j in zip(a, b)] for a, b, in zip(arr1, arr2)]     
    return answer

 

 

 

---

다른 분들 풀이

import numpy as np

def sumMatrix(A,B):
    A_np = np.array(A)
    B_np = np.array(B)
    result = A_np + B_np
    return result.tolist()
    # return [[c + d for c, d in zip(a, b)] for a, b, in zip(A, B)]

# 아래는 테스트로 출력해 보기 위한 코드입니다.
print(sumMatrix([[1,2], [2,3]], [[3,4],[5,6]]))

와 넘파이도 되는구나!

 

파이썬 리스트...배열이 너무 익숙해서 그런건지 자꾸 헷갈린다

오늘 진짜 list comprehension 복습하고 자야지...

흑