顧名思義，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]]])