将特定坐标增加到numpy数组[重复]

Question

我对 numpy 很陌生，我想使用 numpy 增加数组的特定位置（所以它更快）。

为了清楚，我有一个“信息”类数组中的坐标列表。

class info:
    def __init__(self, timestamp, position):
        self.timestamp = timestamp
        self.position = position

目前我这样做：

for data in datas:
    x = data.position[0]
    y = data.position[1]

    self.heatmap[x][y] += 1

“热图”也是一个大小为宽高的numpy数组。

我想知道是否有更“numpy”的方式来做到这一点。

编辑：对不起，我正在寻找的是世界“指数”。答案就在那里：Increment given indices in a matrix

Answer 1

例如，你可以使用这样的东西吗？

import numpy as np
a = np.arange(16).reshape(4,4)
b = [0,0]
c = (1,1)
d = (np.array([2]),np.array([2]))
e = ([3],[3])
a[b] = 100  # b isn't a tuple so it doesn't work the way the others do
a[c] = 200
a[d] = 300
a[e] = 400
print a

[[100 100 100 100]
 [  4 200   6   7]
 [  8   9 300  11]
 [ 12  13  14 400]]

所以你可以考虑

self.heatmap[tuple(data.position)] += 1