基于Numpy元组索引的2d数组加法赋值

Question

如果我们有一个二维数组，例如100x100，并且我们想根据它们的坐标为它添加值，我们怎样才能避免覆盖先前坐标值的值？

import numpy as np

vm_map = np.zeros((100,100))
a = np.array([[0,1], [10,10], [40,40], [40,40]])
vm_map[tuple(a.T)] = vm_map[tuple(a.T)] + [1,.5,.3, .2]

print(vm_map[40,40])

我们希望这个代码块打印.5，添加两个[40,40]坐标，而不是打印.2，因为那是它在该坐标处接收的最后一个值。

Answer 1

您可以使用np.add.at在指定坐标处进行就地添加：

vm_map = np.zeros((100,100))
a = np.array([[0,1], [10,10], [40,40], [40,40]])

np.add.at(vm_map, tuple(zip(*a)), [1,.5,.3, .2])

print(vm_map)
array([[0., 1., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       ...,
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.]])