Мой вопрос касается конкретной операции с массивом, которую я хочу выразить с помощью numpy. .
У меня есть массив чисел с плавающей запятой w
и массив индексов idx
той же длины, что и w
, и я хочу просуммировать все w
с тем же значением idx
и собрать их в массив v
.
В виде цикла это выглядит так:
for i, x in enumerate(w):
v[idx[i]] += x
Есть ли способ сделать это с помощью операций с массивами?
Мое предположение было v[idx] += w
, но это не работает, так как idx
содержит один и тот же индекс несколько раз.
Спасибо!