python - 访问 Numpy 列表中的值范围

Question

假设我有两个包含坐标(point1 和 point2)的大小为 n 的向量，以及一些具有 n 维的大型 Numpy 数组(len(array.shape) == 3)。

现在，point1 的所有值都小于 point2，我想提取 point1 和 point2 之间包含的子数组。如果我事先知道维度数 n(例如 n=3)，我会像这样访问它:

数组[point1[0]:point2[0], point1[1]:point2[1], point1[2]:point2[2]]

我想知道 Numpy 中是否有一种干净的 Pythonic 方法可以适用于任意数量的维度？

谢谢!

Answer 1

array[map(slice,point1,point2)]

A[0:2,0:2] 的索引与 (slice(0,2), slice(0,2)) 相同，其中是一个切片元组。