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

berry ·2021년 6월 14일
0

Algorithm

목록 보기
37/77
post-thumbnail

문제


🧩 numpy

  • import numpy as np
  • 주로 matplotlib과 함께 사용함
  • 벡터와 행렬 연산 주

🏁 내 풀이

import numpy as np

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

📌

  • numpy를 이용해서 array로 연산 (원소합)
  • answer.tolist()를 이용하여 array 👉🏻 list

🧩 다른 풀이

def sumMatrix(A,B):
    answer = [[c + d for c, d in zip(a, b)] for a, b in zip(A,B)]
    return answer

📌

  • zip(A,B)하여 A,B를 a,b로 나눈 후 a,b로 나눈 것을 다시 c,d로 나눈 다음 c+d

🧩 zip()

profile
Engineer

0개의 댓글