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)