Суммарный массив по числу в numpy

Предполагая, что у меня есть массив типа numpy: [1,2,3,4,5,6] и еще один массив: [0,0,1,2,2,1] Я хочу суммировать элементы в первом массиве по группам (второй массив) и получить результаты n-групп в порядке номеров групп (в этом случае результат будет [3, 9, 9]). Как мне это сделать в numpy?