[python]プログラマ(Lv 2)-マトリクス乗算

こんにちは:
この問題は行列乗算のように...解けばいい.
https://programmers.co.kr/learn/courses/30/lessons/12949



ここで、arr 1の行はrであり、行列arr 1の列/arr 2の行はmであり、arr 2の列はnである.

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