Как я могу уменьшить массив numpy в ячейки с максимальным значением для ячейки?

Question

Мой массив выглядит следующим образом: (212960,)

Я хочу разделить это на 625 корзины одинаковой длины. Поскольку 212960/625 не является целым числом, меня не волнуют оставшиеся части.

Для каждого бина я хочу заполнить его любым максимальным значением.

I чувствую, что нужно .reshape, просто не знаю, как это сделать.

Answer 1

Вот что я бы сделал (и да, с reshape):

# sample data
a = np.arange(212960)

bins = 625

max_size = (len(a)//bins) * bins

new_arr = a[:max_size].reshape(bins, -1).max(1)