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]
 
print(slice1)
print(slice2)
print(slice3)
print(slice4)