프로그래머스 LV2 - 행렬의 곱셈

문제 링크

내 풀이

import numpy as np

def solution(arr1, arr2):
    arr1 = np.array(arr1)
    arr2 = np.array(arr2)
    result = arr1.dot(arr2)
    return result.tolist()

다른 풀이

def productMatrix(A, B):
    return [[sum(a*b for a, b in zip(A_row,B_col)) for B_col in zip(*B)] for A_row in A]

풀이 해석

• zip과 sum을 이용해 컴프리핸션으로 간결하게 풀이하였다

배울 점

• list comprehensin의 이용
• numpy의 dot함수