Как суммировать значения на основе второго индексного массива векторизованным способом ⇐ Python

[Anonymous]

Предположим, у меня есть массив значений

Код: Выделить всё

values = np.array([0.0, 1.0, 2.0, 3.0, 4.0])

и индексный массив

Код: Выделить всё

indices = np.array([0,1,0,2,2])

Существует ли векторизованный способ суммирования значений для каждого уникального индекса в индексах?
Я имею в виду векторизованную версию для получения сумм в этом фрагменте :

Код: Выделить всё

sums = np.zeros(np.max(indices)+1)
for index, value in zip(indices, values):
sums[index] += value

Бонусные баллы, если решение позволяет значениям (и, как следствие, суммам) быть многомерными.


Подробнее здесь: https://stackoverflow.com/questions/791 ... zed-manner