-
[Programmers/프로그래머스]행렬의 덧셈코딩테스트/Programmers 2022. 9. 29. 21:54728x90반응형
문제 설명
행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.
제한 조건- 행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.
arr1 arr2 return [[1,2],[2,3]] [[3,4],[5,6]] [[4,6],[7,9]] [[1],[2]] [[3],[4]] [[4],[6]] 풀이
이 문제는 행렬을 다루는 문제다. 파이썬에서 행렬을 다룰때는 행렬 관련 각종 메서드가 있는 numpy 배열로 변환해서 다루는 것이 편하다.
numpy에서 행렬의 덧셈은 numpy 배열끼리 + 연산을 하면된다.
그리고 결국에는 리스트로 반환을 해야되기 때문에 tolist() 메서드를 사용해서 리스트로 변환한 후 반환하면 된다.
import numpy as np def solution(arr1, arr2): return (np.array(arr1)+np.array(arr2)).tolist()
728x90반응형'코딩테스트 > Programmers' 카테고리의 다른 글
[Programmers/프로그래머스]직사각형 별 찍기 (0) 2022.09.29 [Programmers/프로그래머스]부족한 금액 계산하기 (0) 2022.09.29 [Programmers/프로그래머스] 약수의 개수와 덧셈 (2) 2022.09.29 [Programmers/프로그래머스]문자열 다루기 기본 (0) 2022.09.29 [Programmers/프로그래머스]문자열 내림차순으로 배치하기 (0) 2022.09.28