Numpy多维数组切片

import numpy as np
 
# 创建一个3x4x5的多维数组
array = np.arange(60).reshape(3, 4, 5)
 
# 获取第一个维度的所有元素，第二个维度的前两个元素，第三个维度的所有元素
slice1 = array[:, :2, :]
 
# 获取第一个维度的第一个元素，第二个维度的所有元素，第三个维度的前三个元素
slice2 = array[0, :, :3]
 
# 获取第一个维度的前两个元素，第二个维度的间隔取样（比如每隔一个元素取一个），第三个维度的所有元素
slice3 = array[:2:2, :, :]
 
# 获取第一个维度的所有元素，第二个维度的第1、3个元素，第三个维度的间隔取样（比如每隔两个元素取一个）
slice4 = array[:, [0, 2], ::2]
 
slice1
 [[[ 0  1  2  3  4]
  [ 5  6  7  8  9]]

 [[20 21 22 23 24]
  [25 26 27 28 29]]

 [[40 41 42 43 44]
  [45 46 47 48 49]]]
slice2
 [[ 0  1  2]
 [ 5  6  7]
 [10 11 12]
 [15 16 17]]
slice3
 [[[ 0  1  2  3  4]
  [ 5  6  7  8  9]
  [10 11 12 13 14]
  [15 16 17 18 19]]]
slice4
 [[[ 0  2  4]
  [10 12 14]]

 [[20 22 24]
  [30 32 34]]

 [[40 42 44]
  [50 52 54]]]