풀이 과정

  1. numpy의 matmul 함수를 활용하여 행렬의 곱셈 수행

  2. matmul 함수 활용하기 위해 리스트를 np.array로 변환하는 과정 필요

  3. numpy array를 list로 변환하기 위해 tolist 함수 사용

import numpy as np

def solution(arr1, arr2):
    a1 = np.matmul(np.array(arr1),np.array(arr2))
    answer = a1.tolist()
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges

+ Recent posts