programmers | Lv2. 행렬의 곱셈 [Python]

💡 Python 3

🔗 문제

행렬의 곱셈 [Link]

💻 코드

def solution(arr1, arr2):
    import numpy as np
    arr3, arr4 = np.array(arr1), np.array(arr2)
    result = arr3 @ arr4
    return result.tolist()

💥 다른 사람 코드

오늘도 또 하나 배워간다.. numpy를 활용할 수 없는 경우에는 코드를 직접 작성해야하는데 많이 도움 될 것 같다.
*를 언패킹에 사용할 수 있는 것을 처음 알았다!!
zip에 대해서도 다시 한번 알고 갈 수 있어서 좋았다!!

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]

참고 자료

[파이썬]Asterisk(*)의 4가지 기능
[파이썬] 내장 함수 zip 사용법