通过索引和掩码更新？

Question

我有一个大的二维数组，我可以通过索引访问它。我只想更新索引数组中不为零的值。

arrayx = np.random.random((10,10))

假设我有索引（这只是示例，实际索引是由单独的进程生成的）：

idxs = np.array([[4],
        [5],
        [6],
        [7],
        [8]]), np.array([[5, 9]])

鉴于这些索引，这应该可以工作，但它没有。

arrayx[idxs]

array([[0.7 , 0.1 ],
       [0.79, 0.51],
       [0.  , 0.8 ],
       [0.82, 0.32],
       [0.82, 0.89]], dtype=float16)

// note from editor: '<>' is equivalent to '!='
// but I agree that '>' 0 is more correct
// mask = mapx[idxs] <> 0 // original
mask = arrayx[idxs] > 0 // better

array([[ True,  True],
       [ True,  True],
       [False,  True],
       [ True,  True],
       [ True,  True]])

arrayx[idxs][mask] += 1

但是，这不会更新数组。我该如何解决这个问题？

Answer 1

一个简单的 np.where 带有掩码作为第一个输入来选择和分配 -

mapx[idxs] = np.where(mask,mapx[idxs]+1,mapx[idxs])

自定义更新值

第二个参数（此处为mapx[idxs]+1）可以编辑为您可能对True 中mask 中的True 对应的屏蔽位置进行的任何复杂更新。因此，假设您正在对蒙面的地方进行更新：

mapx[idxs] += x * (A - mapx[idxs])

然后，将第二个 arg 替换为 mapx[idxs] + x * (A - mapx[idxs])。

---

另一种方法是从mask 中的True 中提取整数索引，然后根据掩码选择性地创建新的idxs，就像这样 -

r,c = np.nonzero(mask)
idxs_new = (idxs[0][:,0][r], idxs[1][0][c])
mapx[idxs_new] += 1

最后一步可以类似地编辑以进行自定义更新。只需使用idxs_new 代替idxs 即可更新。