[알고리즘]기초1-행렬의 덧셈

행렬의 덧셈

https://programmers.co.kr/learn/courses/30/lessons/12950

✔ 반복문 이용

def solution(arr1, arr2):
    answer = []

    for i in range(len(arr1)):
        temp=[]
        for j in range(len(arr1[i])):
            value=arr1[i][j]+arr2[i][j]
            temp.append(value)
        answer.append(temp)
    return answer

👉🏻 이중 for문과 if를 이용해서 일일이 더해주는 방식. 힌트는 메소드 한 줄로 풀어보기였지만 알고 있는 적절한 메소드가 없어서 이렇게 풀어봤다.

• arr1의 이중 리스트의 길이를 for문의 range로 설정함
• 각 반복마다 arr1[i][j]+arr2[i][j]의 값을 temp.append한다.
• 내부 for문이 끝나면 answer.append(temp)를 하여 answer을 반환

✔ list comprehension 이용

코드를 입력하세요

❗ 파이썬의 기본 문법들 복습을 잘 해야겠다.. list comprehension은 생각도 못했음...!😂

✔ numpy 패키지의 array 이용

 import numpy as np

 def solution(arr1, arr2):
 	answer = [[]]
    arr1_1=np.array(arr1)
    arr2_1=np.array(arr2)
    
    answer=(arr1_1+arr2_1).tolist()

    return answer

👉🏻 문제에서 arr1과 arr2는 행과 열의 크기가 같다고 하였으므로 행렬의 덧셈 메소드를 이용할 수 있다.

• numpy 패키지를 import 후 arr1과 arr2를 각각 array()를 이용하여 행렬로 변환

• arr1과 arr2를 합한 결과를 .tolist()를 이용하여 list로 변환

  ❗ 파이썬에서 주로 쓰이는 패키지들의 주요 메소드들을 파악하고 있어야겠다.