[프로그래머스] 연습문제 - 행렬의 덧셈 문항 (Python)

문제출처 : 프로그래머스

👻 문제

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

• 제한 조건
  행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

• 입출력 예
  

👩‍💻 My cording

풀이 1 : numpy 활용

import numpy as np

def solution(arr1, arr2):
    arr1_np = np.array(arr1)
    arr2_np = np.array(arr2)
    sum = arr1_np + arr2_np
    #tolist()를 해야 numpy의 array 형식 → list로 변환
    return sum.tolist() 

---

💡check point

tolist() : numpy의 array 형식 → list로 변환

풀이 2 : for문과 if문 활용

def solution(arr1, arr2):
  for i in range(len(arr1)): # 0이상 arr1 개수 미만 (전체)
    for j in range(len(arr1[i])):
      arr1[i][j]+=arr2[i][j]
   
  return arr1

---

💡check point

for문 : range = len(list) → 전체 인덱스 범위 (0이상 list 개수 미만)