脑子 发表于 2020-6-5 11:58:07

【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)

_2_ 发表于 2020-6-5 20:22:56

看着很乱……
改一下代码段吧
页: [1]
查看完整版本: 【np.newaxis用法】增加维度