无需循环即可访问3D numpy数组中的多个列

import numpy as np

# 创建一个3D numpy数组
arr = np.array([[[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, 27]]])

# 使用切片操作选择多个列
columns = arr[:, :, [0, 2]]  # 选择第1列和第3列

print(columns)

[[[ 1  3]
  [ 4  6]
  [ 7  9]]

 [[10 12]
  [13 15]
  [16 18]]

 [[19 21]
  [22 24]
  [25 27]]]