a[:, 1] 忽略第一维，第二维取1

：的意思是忽略这一维

例如

a = np.resize(np.arange(27), (3, 3, 3))

[[[ 0  1  2]
  [ 3  4  5]
  [ 6  7  8]]

 [[ 9 10 11]
  [12 13 14]
  [15 16 17]]

 [[18 19 20]
  [21 22 23]
  [24 25 26]]]

其中a的形状是(3,3,3)。

我们取其中一个值，需要给一个三维坐标，如(x,y,z)

a[:, 1] 就是指忽略x，y=1的数。

即a[0,1,0],a[0,1,1],a[0,1,2],a[1,1,0],a[1,1,1],a[1,1,2],a[2,1,0],a[2,1,1],a[2,1,2]

一共9个数

import numpy as np

if __name__ == '__main__':
    a = np.resize(np.arange(27), (3, 3, 3))
    print(a[:, 1])

[[ 3  4  5]
 [12 13 14]
 [21 22 23]]

再如a[:, :, 2]就是前两维都不要，只要最后一维为2

即a[0,0,2],a[0,1,2],a[0,2,2],a[1,0,2],a[1,1,2],a[1,2,2],a[2,0,2],a[2,1,2],a[2,2,2]

import numpy as np

if __name__ == '__main__':
    a = np.resize(np.arange(27), (3, 3, 3))
    print(a[:, :, 2])

[[ 2  5  8]
 [11 14 17]
 [20 23 26]]