Numpy rolling 2d

def rolling_window(a, window_size):
    shape = (a.shape[0] - window_size + 1, window_size) + a.shape[1:]
    strides = (a.strides[0],) + a.strides
    return np.lib.stride_tricks.as_strided(a, shape=shape, strides=strides)

Source: stackoverflow.com

Similar codes

6 votes327 views2 answers
How to run a File using python
Python
0 vote305 views1 answer
Python get list memory size
Python
5 votes314 views6 answers
Joe biden
Python
1 vote320 views1 answer
ImportError: dynamic module does not define module export function
Python
0 vote321 views1 answer
Iterating over the two ranges simultaneously and saving it in database
Python
0 vote354 views2 answers
Python requests json backslash
Python
1 vote370 views1 answer
Sklearn accuracy
Python
4 votes311 views1 answer
Python make integer into a list
Python
1 vote323 views1 answer
Split a variable into multiple variables in python
Python
76 votes366 views6 answers
Python file open
Python