Arrays 如何在选定的维度切片数组？

假设我有一个多维numpy数组A。我尝试在我选择的任何维度中获取A的切片，例如，如果

A

是一个3D数组，并且我被赋予了一些整数

层

和一些维度

dim

：

如果dim=2，请选择一个[：，：，层]

如果dim=1，请选择一个[：，层，：]

如果dim=0，请选择一个[图层，：，：]

---

在我的例子中，A可能有许多维度可供选择，所以我不想手动编写。什么是一种简单易读的方法？

np.take

可能适合您的需要。“像np.take（arr，index，axis=3）这样的调用相当于arr[：，：，：，：，index，…]”哦，这听起来正是我需要的。非常感谢。对不起，我不是想问一个已经有函数的问题。请注意，这只适用于高级索引。它不适用于切片。