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