[파이썬] 행렬의 곱셈 [CODING TEST #31]

https://school.programmers.co.kr/learn/courses/30/lessons/12949

 

 

두 행렬의 곱은 조건이 한정적이어서 간단하게 구현이 가능하다.

곱의 결과는 [arr1의 row x arr2의 column] 형태를 띠며, arr1의 column과 arr2의 row가 같아야 한다.

 

def solution(arr1, arr2):    
    arr1_row, mid, arr2_col = len(arr1), len(arr2), len(arr2[0])
    answer = [[0 for _ in range(arr2_col)] for _ in range(arr1_row) ]
    
    for i in range(arr1_row):
        for j in range(arr2_col):
            for m in range(mid):
                answer[i][j] += arr1[i][m] * arr2[m][j]
    
    return answer