def cumsum_based(bin_edges, n):
    id_arr = np.zeros(n, dtype=int)
id_arr[bin_edges + 1] = 1
return id_arr.cumsum()
Увеличьте элементы массива numpy на 1 по определенным индексам (для использования с группировкой таблицы astropy)