关于python 中np.newaxis的用法

顾名思义，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]]])