如果我们有一个2d数组,例如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,因为这是它在该坐标处收到的最后一个值。
答案 0 :(得分:3)
您可以使用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.]])