Prefix reduce operation on Numpy Array

Asked Apr 12 '20 at 08:37

Active Apr 12 '20 at 09:56

Viewed 462 times

Given an array say a = [2,6,4,1,5] how can I apply a prefix reduce operator on it.

For example:

Prefix sum of a = [2,8,12,13,18] or

Prefix max of a = [2,6,6,6,6]

I want a method using which I can use any suitable reduce operators like mean, max or min on the array.

edited Apr 12 '20 at 09:56

FBruzzesi

asked Apr 12 '20 at 08:37

djin31

3

Are you aware of `ufunc.accumulate`? As in `np.maximum.accumulate(a)`? – Paul Panzer Apr 12 '20 at 08:51
@PaulPanzer Just out of curiosity, any tips on how to use accumulate with mean? – Dani Mesejo Apr 12 '20 at 09:00
@DaniMesejo I can't think of anything better than taking sums and afterwards divide by counts – Paul Panzer Apr 12 '20 at 09:25
@PaulPanzer thanks this is what I was looking for. I kept searching for this functionality in ufunc.reduce and related operators. – djin31 Apr 12 '20 at 10:08

0 Answers0