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

Question

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

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

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

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

Спасибо!

Answer 1

numpy.bincount был введен для этой цели:

tmp = np.bincount(idx, w) 
v[:len(tmp)] += tmp 

Я думаю, как 1,6 вы можете также передать MINLENGTH в bincount.

Answer 2

Это известное поведение и, хотя и несколько неудачное, не имеет обходного решения на уровне numpy. (bincount можно использовать для этого, если вы крутите руку). Выполнение цикла самостоятельно - это ваш лучший выбор.

Обратите внимание, что ваш код может быть немного более ясно, без повторного использования имени w и без введения другого набора индексов, как

for i, w_thing in zip(idx, w): 
    v[i] += w_thing 

Если вам нужно ускорить этот цикл, вы, возможно, спуститься к C. Cython делает это относительно легко.