np.newaxis的用法

np.newaxis

1.等价形式

np.newaixs==None
——参考 这篇博客。

2.用法

因为我们平时都是所用的矩阵乘法都是两个二维的矩阵相乘，而通过
arr=np.array([1,2,3])
得到的是一维数组，它的shape==(3,)，同时arr.ndim==1，所以需要指明，将一维数组reshape成二维的矩阵。
方法是：如果希望将其reshape为shape=(3,1)的矩阵，写法 arr=arr[:,np.newaxis]或者直接 arr=arr[:,None]。
如果希望将其reshape为shape=(1,3)的矩阵，写法 arr=arr[np.newaxis,:]或者直接 arr=arr[None,:]。

希望哪一维是1，就令其为np.newaxis.

重新赋值以后，arr.ndim==2，确定地成为了一个二维数组。之后就可以愉快地进行矩阵乘法了。