Numpy快速条件索引操作

Question

假设您有3D阵列。 对大型数组执行以下操作的最快方法是什么：

If:
i_low < i < i_high and
j_low < j < j_high and
k_low < k < k_high

Then:
change the element value to 1

Answer 1

一种相对快速的方法是使用NumPy的切片：

a[i_low+1:i_high, j_low+1:j_high, k_low+1:k_high] = 1