newaxis的用法以及a[0][:, :, np.newaxis]的含义

import numpy as np
a=np.array([[[[1],[1],[1]],[[2],[2],[2]],[[3],[3],[3]]],[[[4],[4],[4]],[[5],[5],[5]],[[6],[6],[6]]]])
a.shape

(2, 3, 3, 1)

b=a[0][:, :, np.newaxis]#np.newaxis是新建一个维度
print(b)
b.shape

[[[[1]]

  [[1]]

  [[1]]]


 [[[2]]

  [[2]]

  [[2]]]


 [[[3]]

  [[3]]

  [[3]]]]





(3, 3, 1, 1)

c=a[0][:,:] a[0]对a的第一个维度只取其中一个值,这样第一个维度相当于没有了
print(c)
c.shape

[[[1]
  [1]
  [1]]

 [[2]
  [2]
  [2]]

 [[3]
  [3]
  [3]]]





(3, 3, 1)

a1=np.array([[1],[2]])
print(a1)
a1.shape

[[1]
[2]]

(2, 1)

b1=a1[0][:] 取a1的第一个维度只取其中一个,这样第一个维度相当于没有了
print(b1)
b1.shape

[1]





(1,)

附上另一个连接：
关于np.newaxis的一点理解：https://blog.csdn.net/molu_chase/article/details/78619731