[코딩테스트 입문] 행렬의 덧셈(Python3)

[코딩테스트 입문] 행렬의 덧셈(Python3)

---

https://school.programmers.co.kr/learn/courses/30/lessons/12950?language=python3 

프로그래머스

 

numpy써서 할까하다가 그냥 했다.

def solution(arr1, arr2):
    answer = [] 
    for i in range(len(arr1)):
        f = []
        for j in range(len(arr1[i])):
            f.append(arr1[i][j]+arr2[i][j])
        answer.append(f)
    return answer

 

다른사람 코드

zip함수ㅎㅎ

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

numpy사용

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