날짜: 2021년 5월 17일
소요 시간: 6분 11초
카테고리: 행렬
태그: 레벨1, 파이썬

코딩테스트 연습 - 행렬의 덧셈

입출력 예시

내가 적은 코드

import numpy as np

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

풀이 과정

행렬의 연산이 가능한 numpy라이브러리를 가져와 문제를 풀었다.

베스트 코드

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

반성

• numpy라이브러리를 알고 있어서 가져와 풀긴 했지만 출제자의 의도는 베스트 코드가 아니었을까…
• 마지막 tolist()를 붙이지 않아 행렬 형태로 답이 출력되는 오류 때문에 시간이 많이 걸렸다.