[Numpy-discussion] Rolling window (moving average, moving std, and more)

Erik Rigtorp erik at rigtorp.com
Mon Jan 3 08:37:21 EST 2011


On Mon, Jan 3, 2011 at 05:13, Sebastian Haase <seb.haase at gmail.com> wrote:
> Hi Erik,
> This is really neat !  Do I understand correctly, that you mean by
> "stride tricks", that your rolling_window is _not_ allocating any new
> memory ?
Yes, it's only a view.

> IOW, If I have a large array using 500MB of memory, say of float32 of
> shape 125,1000,1000 and I want the last axis rolling of window size
> 11, what would the peak memory usage of that operation be ?
It's only a view of the array, no copying is done. Though some
operations like np.std()  will copy the array, but that's more of a
bug. In general It's hard to imagine any speedup gains by copying a
10GB array.

> How about renaming the option `window` to `window_size`  (first I was
> thinking of things like hamming and hanning windows...)... ?
Sounds fare.



More information about the NumPy-Discussion mailing list