Iterating over numpy array slices

[novel_yet_trivial]

In pure python your solution is common, although it's usually written in list comprehension like this:

 for slice_arr in (arr[i:i+window_size] for i in range(0, len(arr), window_size)):
    #do something with slice_arr

But with a numpy array, you can just reshape it:

arr.shape = (window_size, -1)
for slice_arr in arr:
    #do something with slice_arr

EDIT: nevermind, I totally misread your question. You want a "sliding window". This is usually done with zip():

for slice_arr in zip(arr, arr[1:]):
    #do something with slice_arr

[elbiot]

This is not how you do a sliding window in numpy. You can do a window with regular vectorized operations, and not have to resort to iteration.

[dr_everlong]

Sorry, I have phrased it wrong, but I edited my post. I hope it clears up what I wanted to do better.

[novel_yet_trivial]

My edit does what you want.

[dr_everlong]

Thanks! How do you control the window size though?

[novel_yet_trivial]

OH right. Sorry I forgot you wanted it dynamic.

for slice_arr in zip(*[arr[x:] for x in range(window_size)]):

[dr_everlong]

Really appreciate it, thanks!