[프로그래머스 파이썬] 행렬의 곱셈

📌문제 출처

프로그래머스 LV 2
https://school.programmers.co.kr/learn/courses/30/lessons/12949

프로그래머스

❓ 문제

📗 풀이 코드

def solution(arr1, arr2):
    ans = [[0]*len(arr2[1]) for _ in range(len(arr1))]
    for i in range(len(arr1)):
        for j in range(len(arr2[1])):
            row = arr1[i]
            col = [row2[j] for row2 in arr2]
            tmp = 0
            for k,l in zip(row,col):
                tmp += k*l
            ans[i][j] = tmp
    return ans