프로그래머스 | 행렬 곱(python)

stillssi·2023년 2월 28일
0

코테 복습하기

목록 보기
7/15

프로그래머스Lv.2 행렬의 곱셈

def solution(arr1, arr2):
    answer = []

    for i in range(len(arr1)):
        arr = []
        for j in range(len(arr2[0])):
            sum = 0
            for k in range(len(arr1[0])):
                sum += arr1[i][k] * arr2[k][j]
            arr.append(sum)
        answer.append(arr)

    return answer
       

행렬 [2*3][3* 4]의 곱의 결과 행렬 = [2*4]

첫번째 for문 1번째 행렬의 행
두번째 for문 2번째 행렬의 열
==> 결과 행렬의 모양

0개의 댓글