관리 메뉴

솜씨좋은장씨

[Programmers] 행렬의 곱셈 (Python) 본문

Programming/코딩 1일 1문제

[Programmers] 행렬의 곱셈 (Python)

솜씨좋은장씨 2021. 5. 7. 01:17
728x90
반응형

코딩 1일 1문제 오늘의 문제는 프로그래머스의 행렬의 곱셈 입니다

 

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

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

Solution

import numpy as np

def solution(arr1, arr2):
    answer = [[]]
    
    answer = (np.matrix(arr1)*np.matrix(arr2)).tolist()
    
    return answer

Solution 풀이

numpy의 matrix를 활용하여 각각의 리스트를 행렬로 만들어준 다음 이를 곱한다음 

다시 리스트로 변환한 값을 정답으로 return합니다.

 

SOMJANG/CODINGTEST_PRACTICE

1일 1문제 since 2020.02.07. Contribute to SOMJANG/CODINGTEST_PRACTICE development by creating an account on GitHub.

github.com

Comments