【np.newaxis用法】增加维度
>>> np.linspace(1,-1,5)array([ 1. ,0.5,0. , -0.5, -1. ])
#=============================================================
>>> np.linspace(1,-1,5)[:,np.newaxis]
array([[ 1. ],
[ 0.5],
[ 0. ],
[-0.5],
[-1. ]])
>>> np.linspace(1,-1,5)[:,np.newaxis].shape
(5, 1)
#==============================================================
>>> np.linspace(1,-1,5)
array([[ 1. ,0.5,0. , -0.5, -1. ]])
>>> np.linspace(1,-1,5).shape
(1, 5)
看着很乱……
改一下代码段吧
页:
[1]