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

Question

假设我有一个3D 100x100x100阵列，并且有一些起点和终点坐标，例如：

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

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

Answer 1

我们可以使用切片表示法。对于这个具体案例：

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))]