S E P H ' S

[Python] 행렬의 곱셈 본문

Algorithm/Programmers

[Python] 행렬의 곱셈

yoseph0310 2021. 7. 20. 21:35
 

코딩테스트 연습 - 행렬의 곱셈

[[2, 3, 2], [4, 2, 4], [3, 1, 4]] [[5, 4, 3], [2, 4, 1], [3, 1, 1]] [[22, 22, 11], [36, 28, 18], [29, 20, 14]]

programmers.co.kr

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

'Algorithm > Programmers' 카테고리의 다른 글

[Python] 최댓값과 최솟값  (0) 2021.07.31
[Python] 최솟값 만들기  (0) 2021.07.31
[Python] 키패드 누르기  (0) 2021.07.18
[Python] 숫자 문자열과 영단어  (0) 2021.07.09
[Python] 2019 KAKAO BLIND RECRUITMENT 실패율  (0) 2021.07.08