从开始/结束坐标获取 NumPy 切片

Question

假设我有一个 3D 100x100x100 数组并且有一些起始和结束坐标，例如：

(0,0,0) (50,20,20)

我如何获取 提取一个由这些顶点定义的长方体包围的子数组？

Answer 1

我们可以使用 slicing 表示法。对于这种特定情况，这将是：

subarray = the_array[0:50, 0:20, 0:20]

当然，只有在我们事先知道数字（和尺寸）的情况下，上述方法才有效。如果给定两个三元组，则需要一个更复杂的程序：

first = (0,0,0)
second = (50,20,20)

a0, b0, c0 = first
a1, b1, c1 = second
subarray = the_array[a0:a1, b0:b1, c0:c1]

但这仍然不是很优雅，因为我们硬编码了维数。为此，我们可以处理两个任意大小的元组（或一般的可迭代对象）：

first = (0,0,0)
second = (50,20,20)

subarray = the_array[tuple(slice(x,y) for x, y in zip(first, second))]

切片的上限总是互斥的。如果你想添加这些，你可以增加上限：

subarray = the_array[0:51, 0:21, 0:21]

或：

subarray = the_array[a0:a1+1, b0:b1+1, c0:c1+1]

或：

subarray = the_array[tuple(slice(x,y+1) for x, y in zip(first, second))]