Numpy накапливает один массив в другом, используя массив индексов

Мой вопрос касается конкретной операции с массивом, которую я хочу выразить с помощью numpy. .

У меня есть массив чисел с плавающей запятой wи массив индексов idxтой же длины, что и w, и я хочу просуммировать все wс тем же значением idxи собрать их в массив v. В виде цикла это выглядит так:

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

Есть ли способ сделать это с помощью операций с массивами? Мое предположение было v[idx] += w, но это не работает, так как idxсодержит один и тот же индекс несколько раз.

Спасибо!

11
задан Andreas Mueller 20 March 2012 в 16:22
поделиться