[Programmers] Lv2. 行列の乗算


問題の説明
2 Dマトリクスarr 1とarr 2を入力し、arr 1にarr 2を乗じた結果の関数を返します.
せいげんじょうけん
行列arr 1、arr 2の行および列の長さは、2または100を超えない.
行列arr 1、arr 2の要素は、−10または20以下の自然数である.
乗算できる配列のみ.
に答える
def solution(arr1, arr2):
    answer = []
    
    for elem1 in arr1:
        tmp = []
        
        for elem2 in zip(*arr2):
            result = sum([e1 * e2 for e1, e2 in zip(elem1, elem2)])
            tmp.append(result)
        answer.append(tmp)
        
    return answer