data = np.arange(1, 7)
groups = np.array([0, 0, 1, 2, 2, 1])
unique_groups = np.unique(groups)
sums = []
for group in unique_groups:
    sums.append(data[groups == group].sum())
Массив суммы по номеру в numpy