[python] 행렬의 곱셈

https://programmers.co.kr/learn/courses/30/lessons/12949?language=python3 

코딩테스트 연습 - 행렬의 곱셈

행렬의 곱셈은 위를 참조.

 

def solution(arr1, arr2):
    
    a1 = len(arr1)
    a2 = len(arr2[0])
    answer = [[0]*a2 for _ in range(a1)]
    for i in range(a1):
        for j in range(a2):
            for k in range(len(arr2)):
                answer[i][j] += arr1[i][k]*arr2[k][j]
    return answer

i와 j의 값이 고정되어 있는 동안 k의 값이 변하며 곱셈, 덧셈을 수행한뒤 저장하고 i,j의 값을 변경한다.