基於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.]])