python - 如何在 Python 中创建矩阵或将二维数组转换为矩阵？

Question

我希望能够从 Python 中的 2D 数组中提取行或列，以便它保留 2D 形状并可用于矩阵乘法。但是，我在 documentation 中找不到如何才能最好地做到这一点。例如，我可以使用

a = np.zeros(shape=(6,6)) 

创建一个数组，但 a[:,0] 将具有 (6,) 的形状，并且我无法将其乘以形状 (6,1) 的矩阵。我是否需要为每个矩阵乘法将数组的行或列 reshape 为矩阵，或者是否有其他方法来进行矩阵乘法？

Answer 1

您可以直接使用np.matrix:

>>> a = np.zeros(shape=(6,6)) 
>>> ma = np.matrix(a)
>>> ma
matrix([[ 0.,  0.,  0.,  0.,  0.,  0.],
        [ 0.,  0.,  0.,  0.,  0.,  0.],
        [ 0.,  0.,  0.,  0.,  0.,  0.],
        [ 0.,  0.,  0.,  0.,  0.,  0.],
        [ 0.,  0.,  0.,  0.,  0.,  0.],
        [ 0.,  0.,  0.,  0.,  0.,  0.]])
>>> ma[0,:]
matrix([[ 0.,  0.,  0.,  0.,  0.,  0.]])

或者您可以使用np.newaxis添加尺寸

>>> a[0,:][np.newaxis, :]
array([[ 0.,  0.,  0.,  0.,  0.,  0.]])