💡 문제

행렬의 곱셈

• 유형 연습문제

• 문제 설명 2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

• 제한 조건
  • 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
  • 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
  • 곱할 수 있는 배열만 주어집니다.

• 입력
  • ex1) [[1, 4], [3, 2], [4, 1]], [[3, 3], [3, 3]]
  • ex2) [[2, 3, 2], [4, 2, 4], [3, 1, 4]], [[5, 4, 3], [2, 4, 1], [3, 1, 1]]

• 출력
  • ex1) [[15, 15], [15, 15], [15, 15]]
  • ex2) [[22, 22, 11], [36, 28, 18], [29, 20, 14]]

✍🏻 풀이

SOL 1

# SOL 1)

import numpy as np 

def solution(arr1, arr2):
    return (np.dot(np.array(arr1), np.array(arr2))).tolist()

numpy의 array 함수에 리스트를 넣으면 ndarray 클래스 객체인 배열로 변환해준다. arr1, arr2를 배열로 변환해준 후 행렬곱 함수인 np.dot()을 이용해서 연산해주면 된다. return시에는 출력 형식을 맞춰주기 위해 리스트로 다시 변환해주었다.

🌵 정리

깃헙에서 전체 코드 확인하기