如何在选定的维度上对数组进行切片？

问：

假设我有一个多维 numpy 数组 A。我正在尝试在我选择的任何维度上获得 A 的切片，例如，如果是一个 3D 数组，并且我得到了一些整数和一些维度：Alayerdim

如果 dim = 2，则选择 A[：，：，layer]

如果 dim = 1，则选择 A[：，layer，：]

如果 dim = 0，则选择 A[layer，：，：]

就我而言，A 可能有很多维度可供选择，因此我不想手动编写。什么是简单易读的执行此操作的方法？