我有一个大的二维数组,通过索引访问。我想仅更新非零索引数组的值。
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
然而,这不会更新数组。我该如何解决这个问题?
map
作为变量名,因为它会覆盖 Python 的内置map
函数。 - tedmapx
本身吗? - Mad Physicist