python - NumPy 中的矩阵乘法行为

Question

有以下几点：

In [14]: A = array([[1, 1], [3, 2], [-4, 1]])

In [15]: A
Out[15]:
array([[ 1,  1],
       [ 3,  2],
       [-4,  1]])

In [16]: x = array([1, 1])

In [17]: x
Out[17]: array([1, 1])

In [18]: dot(A, x)
Out[18]: array([ 2,  5, -3])

我期待一列，因为 dot() 函数被描述为普通的矩阵乘法。

为什么它返回一行呢？这种行为似乎非常令人沮丧。

Answer 1

x一维向量，因此不知道它是行向量还是列向量。的结果也是如此dot(A, x)。

变成x二维数组，一切都会好的：

In [7]: x = array([[1], [1]])

In [8]: x
Out[8]: 
array([[1],
       [1]])

In [9]: dot(A, x)
Out[9]: 
array([[ 2],
       [ 5],
       [-3]])

最后，如果您更喜欢使用更自然的矩阵表示法，请转换A为numpy.matrix：

In [10]: A = matrix(A)

In [11]: A * x
Out[11]: 
matrix([[ 2],
        [ 5],
        [-3]])