파이썬 행렬 곱셈 예제

문제 내용

다음의 A*B  행렬곱셈식을 파이썬으로 나타내어라.

 

문제 풀이

import numpy as np

A = np.array([[1, 2, 3], [5, 3, 1]])
B = np.array([[-1, 2], [3, 4], [5, -4]])

print(np.dot(A, B))

 

코드 결과

[[20 -2]
 [ 9 18]]

---