python - Numpy 使用索引数组在另一个数组中累积一个数组

Question

我的问题是关于我想用 numpy 表达的特定数组操作。

我有一个 float 组 w 和一个与 w 长度相同的索引数组 idx 我想总结所有w 具有相同的 idx 值，并将它们收集在数组 v 中。作为一个循环，这看起来像这样:

for i, x in enumerate(w):
     v[idx[i]] += x

有没有办法用数组操作来做到这一点？我的猜测是 v[idx] += w 但这不起作用，因为 idx 多次包含相同的索引。

谢谢!

Answer 1

numpy.bincount 是为此目的引入的:

tmp = np.bincount(idx, w)
v[:len(tmp)] += tmp

我认为从 1.6 开始，您还可以将最小长度传递给 bincount。