如何从掩码数组的索引中获取二维数组的索引

Question

我需要一个 2D 数组的原始索引在一个循环中覆盖一个屏蔽数组的索引。我试试：

for i in range(Arr[mask].size):
    tmp = numpy.zeros(mask.shape, dtype=bool)
    tmp[mask][i] = True
    j,k = numpy.where(tmp)
    ...

不幸的是，最后一行返回

(array([], dtype=int64), array([], dtype=int64))   

找到j和k的好方法是什么？

Answer 1

因为它是一个二维数组，你可以简单地遍历掩码而不是遍历掩码数组：

for j, k in zip(*np.where(mask)):
    value = Arr[j,k]
    # ...

如果您还想获取当前（屏蔽）值的编号：

for i, (j, k) in enumerate(zip(*np.where(mask))):
    value = Arr[j,k]
    # ...