我有一个非常简单的功能,如下所示
def new_price(A, B, x):
return np.linalg.inv(A @ B) @ x
这些是我给它的输入
A = np.array([
[2, 0, 1, 0],
[1, 1, 1, 1],
[0, 0, 0, 10]
])
B = np.array([
[3, 3, 3],
[2, 0, 8],
[0, 5, 3],
[0, 0, 10]
])
x = np.array([ 84, 149, 500])
这将返回数组[ 1. 3. 5.]
。但是,当我进行以下相等性检查时,它返回False
v1 = new_price(A, B, x)
v2 = np.array([1.0, 3.0, 5.0])
np.array_equal(new_price(A, B, [ 84, 149, 500]), np.array([1.0, 3.0, 5.0]))
我检查了一下,两个数组的形状和类型是相同的。我在这里缺少什么?