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

2 votes201 views2 answers
Add time to datetime python
Python
20 votes188 views3 answers
Convert a number column into datetime pandas
Python
0 vote174 views1 answer
How to extract numbers from a list in python
Python
0 vote182 views1 answer
Additon of multiple duration timestamps python
Python
0 vote203 views1 answer
Python qt grid span
Python
35 votes180 views2 answers
How to run cmd line commands in python
Python
0 vote240 views1 answer
Django 3.2 compatible to python 3.10?
Python Django
1 vote208 views1 answer
Requirements.txt flask
Python
0 vote176 views1 answer
How to check if a function false python
Python
-2 votes171 views1 answer
Sort function in pandas dataframe to sort specific properties
Python