numpy_groupies_aggregate_groupby_sum_basic_example.py

1import numpy as np
2import numpy_groupies as npg
3
4group_idx = np.array([0, 0, 0, 1, 1, 1])
5a = np.array([1, 2, 3, 4, 5, 6])
6
7# Calculate the sum of elements in 'a' for each group in 'group_idx'
8res = npg.aggregate(group_idx, a, func='sum')
9print(res)
10# Output: [ 6 15]