python - 如何快速屏蔽阵列中的不同切片？

Question

我有一个 3d 数组，其中所有轴的长度都相同(例如 (5,5,5))。我需要屏蔽所有数组，并按照下面的代码保持数组中的某些切片不被屏蔽。我设法使用 for 循环完成了此操作，但我想知道是否有更快的解决方案。

array = np.reshape(np.array(np.random.rand(125)), (5,5,5))
array = ma.array(array, mask=True)
for i in range(array.shape[0]):
    for j in range(array.shape[1]):
        array[i, j, :].mask[i:j] = False

这允许我将这个数组与另一个相同大小的数组相加，同时忽略屏蔽值。

Answer 1

您可以使用广播一步创建整个 mask :

i, j, k = np.ogrid[:5, :5, :5] 
mask = (i>k) | (k>=j)