numpy的newaxis

newaxis的作用为给numpy数组增加新的维度
示例代码：

import numpy as np

a = np.array([1,2,3]) #a.shsape=(3,)
a1 = a[np.newaxis,:]  #a1.shape=(1,3)
print(a1) #[[1 2 3]]
a2 = a[:,np.newaxis]  #a2.shape=(3,1)
print(a2)

打印结果：

[[1 2 3]]
[[1]
 [2]
 [3]]

总结：np.newaxis放在哪个维度上，那个维度就增加1