상상쓰

[프로그래머스] 행렬의 곱셈 본문

Coding Test

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

상상쓰 2021. 7. 6. 17:35

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

 

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

[[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

 

numpy 가 통하는지 확인해봤다. 된다.

 

import numpy as np

def solution(arr1, arr2):
    answer = np.array(arr1).dot(np.array(arr2))
    
    return answer.tolist()
    
print(solution([[1, 4], [3, 2], [4, 1]], [[3, 3], [3, 3]])) # [[15, 15], [15, 15], [15, 15]]

'Coding Test' 카테고리의 다른 글

[백준] Passport Control  (0) 2021.07.08
[프로그래머스] 단속카메라  (0) 2021.07.07
[백준] 포도주 시식  (0) 2021.07.05
[백준] 한조서열정리하고옴ㅋㅋ  (0) 2021.07.05
[백준] 과제  (0) 2021.07.02
'Coding Test' Related Articles more
Comments