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

프로그래머스 '행렬의 덧셈' 풀이

 

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

 

나의 풀이 )

첫번째 풀이

 

아래 더보기란에 추가 설명

1. arr1과 arr2와 똑같은 크기의 answer를 만들어 줍니다.

2. 중첩 for문을 이용하여 각각의 리스트 인덱스에 맞춰 arr1과 arr2의 합을 구해 answer 리스트의 해당 인덱스 값에 넣어줍니다.

 

첫 번째 풀이보다 좀 더 간단하게 풀 수 있는 numpy를 이용한 풀이도 함께 첨부합니다!

 

numpy를 이용한 풀이

 

아래 더보기란에 추가 설명

1. 우선 import를 이용해 numpy를 np로 별명을 지어 불러옵니다.

2. arr1과 arr2를 np.array를 이용해 배열로 바꾼 후 둘을 더하여 answer에 넣어줍니다.

3. answer를 그대로 출력하면 행렬의 형태로 나오기 때문에 출력 시 tolist()를 사용해줍니다.

수정할 부분이 발견되거나 궁금한 점이 있으면 댓글 남겨주세요.

성장을 위한 댓글은 환영입니다 :)