[Programmers / Level2] 행렬의 곱셈

코딩테스트 연습 - 행렬의 곱셈 | 프로그래머스 스쿨

프로그래머스

---

<내 코드>

def solution(arr1, arr2):
    # 결과 행렬의 크기를 arr1의 행 수와 arr2의 열 수로 설정
    result = [[0] * len(arr2[0]) for _ in range(len(arr1))]

    # 행렬 곱셈 수행
    for i in range(len(arr1)):
        for j in range(len(arr2[0])):
            for k in range(len(arr1[0])):
                result[i][j] += arr1[i][k] * arr2[k][j]

    return result

-> numpy 너무 마려운 문제였다.. 행렬 곱셈을 위해 요소마다 접근해서 곱해주는게 너무 번거롭다.

 

<정해코드>

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]