python - 使用数组作为索引增加 numpy 数组元素

Question

我正在尝试有效地更新 numpy 数组 A 的一些元素，使用另一个数组 b 来指示要更新的 A 元素的索引。但是 b 可以包含被忽略的重复项，而我想被考虑在内。我想避免循环 b。为了说明这一点:

>>> A = np.arange(10).reshape(2,5)
>>> A[0, np.array([1,1,1,2])] += 1
>>> A
array([[0, 2, 3, 3, 4],
       [5, 6, 7, 8, 9]])

而我希望输出为:

array([[0, 3, 3, 3, 4],
       [5, 6, 7, 8, 9]])

有什么想法吗？

Answer 1

要正确处理重复索引，您需要使用 np.add.at 而不是 +=。因此，要更新 A 的第一行，最简单的方法可能是执行以下操作:

>>> np.add.at(A[0], [1,1,1,2], 1)
>>> A
array([[0, 4, 3, 3, 4],
       [5, 6, 7, 8, 9]])

可以找到ufunc.at 方法的文档here .