본문 바로가기
2021/코딩테스트

[프로그래머스] Lv.1 행렬의 덧셈

by 에이도 2020. 12. 3.

이번 문제는 행렬의 덧셈입니다!

 

파이썬에서는 행렬을 pandas와 numpy를 사용해서 구현할 수 있습니다.

저는 이번에  numpy를 사용해서 구현해보았습니다!

 

import numpy as np

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

numpy의 array를 통해 입력받은 리스트의 형태를 행렬 형태로 바꾸고, 

덧셈 후 다시 리스트 형태로 바꾸도록 구성해보았습니다!

 

코딩 테스트 문제를 풀며 이번에 처음으로 모듈을 사용해보았습니다. 

파이썬의 가장 큰 장점으로 생각이 되는 모듈을 많이 이용해야겠다고 느꼈습니다.

 

하나하나 코딩하는 노가다도 중요하지만..! 시험은 어쨌든 통과하는 것도 중요하기 때문입니다..! 

 

혹시 numpy에 대해 처음 보시는 분을 위해 간단하게 정리해보았습니다! 

 

댓글