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

문제 설명

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

---

제한 조건

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

---

입출력 예

arr1	arr2	return
[[1, 4], [3, 2], [4, 1]]	[[3, 3], [3, 3]]	[[15, 15], [15, 15], [15, 15]]
[[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]]

---

def solution(arr1, arr2):
    answer = []
    new_arr2 = []
    
    #arr2의 행과 열 변경해주기
    for a2 in arr2:
        for i, a in enumerate(a2):
            if len(new_arr2) <= i:
                new_arr2.append([])
            new_arr2[i].append(a)

    #arr1의 모든 행 대해 arr2의 (기존에 열이었던) 행들과 곱셈 진행
    for a1 in arr1:
        sums = []
        for a2 in new_arr2:
            sum_ = 0
            for i in range(len(a1)):
                sum_ += a1[i] * a2[i]
            sums.append(sum_)
        answer.append(sums)
    return answer

설이니까 양치기 봐주세요 ...

정말 .... 본가 다녀오느라 너무 피곤합니다...............

 

이번 문제는 transpose를 알았더라면 좋았을 문제. 

numpy를 썼으면 되었을 일이지만 사실 그럴거면 np.matrix(A) * np.matrix(B) 해서 곱하면 끝나는 문제였다...

ㅋㅋㅋㅋㅋㅋ

그럼 공부가 안되었겠죠

 

근데 아주 쉽게 transpose 할 수 있던 방법이 있었따.....

 

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]

여기서 arr2 = [ [1, 2], [3, 4] ]일 경우

*arr2 = [1, 2], [3, 4] 처럼 각 원소가 나오게 되고

zip(*arr2) = (1, 3) (2, 4) 가 된다.

[list(a) for a in zip(*arr2)] 이렇게 하면

[ [1, 3], [2, 4] ] 이렇게.. 아주 쉽게 transpose 가 된다

 

숏코딩은 어려웠더라도 저런 방식은 생각해 낼 수 있었다면 좋았을텐데 아주 .. 아쉽고 ...

한 문제를, 쉬운 문제를 풀더라도 좀 더 파이써닉하게 풀 수 있다면 좋을 것 같다