Level2 행렬의 곱셈

def solution(arr1, arr2):
    answer = []
    sum1=0
    n = len(arr1); m = len(arr2); p = len(arr2[0])
    a = []
    for i in range(n):
        for k in range(p):
            for j in range(m):
                sum1 += arr1[i][j]*arr2[j][k]
            a.append(sum1)  
            sum1=0
        answer.append(a); a= []
            
    return answer

 

<테스트 케이스 추가>

 

arr1 : [[1, 2, 3], [4, 5, 6]]

arr2 : [[1, 4], [2, 5], [3, 6]]

return : [[14, 32], [32, 77]]

 

행렬의 곱셈 특징

(2X3) * (3X2) = (2X2)