给定索引列表，如何在没有 for 循环的情况下进行多次添加操作？

Question

标题可能有点糊涂，我在这里解释一下：给定一个数组：arr = np.array([1.0, 2.0])并给出一个索引列表：idxes = np.array([0, 0, 1, 0])以及要添加的号码列表：e = np.array([0.1, 0.35, 0.5, 0.2])我想要的是得到这样的结果：arr[idxes] += e意思是[arr[0], arr[0], arr[1], arr[0]] + [0.1, 0.35, 0.5, 0.2]所以arr[0]加了三倍，arr[0] = arr[0] + 0.1 + 0.35 + 0.2 = 1.65加arr[1]了一次arr[1] = arr[1] + 0.5 = 2.5，我怎么能在没有for循环的情况下直接做到这一点？

Answer 1

使用numpy.add.at：

np.add.at(arr, idxes, e)
arr

输出：

array([1.65, 2.5 ])