如何在NumPy中对基于2D数组的3D数组进行索引？

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
arr_3d = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])

# 选择第一个2D数组的第一行和第二行
rows = np.array([0, 1])
# 选择第一个2D数组的第一列和第三列
cols = np.array([0, 2])

# 使用整数数组索引对3D数组进行索引
result = arr_3d[rows[:, np.newaxis], rows, cols]
print(result)

[[[1 3]
  [4 6]]

 [[7 9]
  [10 12]]]

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
arr_3d = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])

# 创建一个与3D数组形状相同的布尔数组
mask = np.array([[True, False], [False, True]])

# 使用布尔数组索引对3D数组进行索引
result = arr_3d[mask]
print(result)

[1 4 11]