将两个矩阵(2x1)和(2x2(

时间:2018-11-06 02:34:23

标签: python matrix

您好,我的代码必须将点/向量(1,0)乘以矩阵[1.00583,-0.087156],[0.087156,1.00583]。结果应该给我一个新点(x,y) 这是我到目前为止的内容:

import matplotlib.pyplot as plt
import numpy as np
A = np.array([[1],[0]])
B = np.array([[1.00583, -0.087156], [0.087156, 1.00583]])
test =np.multiply(A, B)
print (test)

结果仍然给我一个(2x2)矩阵,而不是我可以用作点的(2x1)矩阵。还有其他功能或更好的方法吗?

2 个答案:

答案 0 :(得分:0)

test =np.matmul(B,A)

这应该可以解决问题。

答案 1 :(得分:0)

首先,如果要进行矩阵乘法,请使用numpy.matmul@运算符,例如B@A

此外,当您定义A

A = np.array([[1],[0]])

这将创建2x1向量(而不是1x2)。因此,如果您想将向量A与矩阵B(2x2)相乘,则应为C = B * A,其中C将为2x1向量

C = B@A

否则,如果要乘以A * B并且B仍然是2x2矩阵,则应将A定义为1x2向量:

A = np.array([1,0])

并获得1x2的结果

C = A@B