顾名思义,np.newaxis的作用就是选取部分的数据增加一个维度,如用创建如下一个4*4的数组
array=random.rand(4,4)
输出为
array([[0.45284467, 0.27883581, 0.72870975, 0.03455946],
[0.74005136, 0.52413785, 0.78433733, 0.80114353],
[0.16559874, 0.56112999, 0.18464461, 0.38968731],
[0.05684794, 0.50929997, 0.45789637, 0.63199181]])
使用np.newaxis
>>> array_add_aix=array[:,np.newaxis]
输出结果为:
array([[[0.45284467, 0.27883581, 0.72870975, 0.03455946]],
[[0.74005136, 0.52413785, 0.78433733, 0.80114353]],
[[0.16559874, 0.56112999, 0.18464461, 0.38968731]],
[[0.05684794, 0.50929997, 0.45789637, 0.63199181]]])
以上是默认选取全部的数据进行增加维度,还可以选取部分的数据增加维度:
>>> array_add_axis=array[0:2,np.newaxis]
>>> array_add_axis
array([[[0.45284467, 0.27883581, 0.72870975, 0.03455946]],
[[0.74005136, 0.52413785, 0.78433733, 0.80114353]]])