[Python] 프로그래머스 코딩테스트 연습 level2 (행렬의 곱셈)

1. 문제 설명

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

 

 

 

 

 

2. 제한 사항

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

 

 

 

 

 

 

3. 내 풀이

import numpy as np
def solution(arr1, arr2):
    answer = []
    arr2_t = np.array(arr2).T
    arr2_t = [[int(i) for i in x] for x in arr2_t]
    
    for i in range(len(arr1)):
        a = []
        for j in range(len(arr2_t)):
            a.append(sum([x*y for x,y in zip(arr1[i],arr2_t[j])]))
        answer.append(a)    
    return answer

 

1. arr2 전치행렬로 바꿔줌

2. np.array를 사용하여 변환했기 때문에 정수로 바뀌지 않아 리스트 컴프리헨션을 사용해서 숫자를 정수로 바꿔줌

3. arr1의 i 번째 원소와 arr2를 전치시킨 리스트를 곱해줌

 

 

 

 

 

4. 다른 사람 풀이

def productMatrix(A, B):
    return [[sum(a*b for a, b in zip(A_row,B_col)) for B_col in zip(*B)] for A_row in A]