python np.newaxis 用法

np.newaxis先明确功能：增加一个维度

in1:

a=np.array([1,2,3,4,5])
print (a.shape)
print (a)

out1:

(5,)
[1 2 3 4 5]

in2:

a=np.array([1,2,3,4,5])
b=a[np.newaxis,:]
print (a.shape,b.shape)
print (a)
print (b)

out2：

(5,) (1, 5)
[1 2 3 4 5]
[[1 2 3 4 5]]

in3：

a=np.array([1,2,3,4,5])
b=a[:,np.newaxis]
print (a.shape,b.shape)
print (a)
print (b)

out3:

(5,) (5, 1)
[1 2 3 4 5]
[[1]
 [2]
 [3]
 [4]
 [5]]

总结

• 数据往往是一维，不能满足矩阵相乘的前提条件，以此需要np.newaxis
• np.newaxis在左，称”1行n列“，在右，称”n行1列“（在哪哪为1）