Rolling window 2d array

# Rolling window for 2D arrays in NumPy
import numpy as np

def rolling_window(a, shape):  # rolling window for 2D array
    s = (a.shape[0] - shape[0] + 1,) + (a.shape[1] - shape[1] + 1,) + shape
    strides = a.strides + a.strides
    return np.lib.stride_tricks.as_strided(a, shape=s, strides=strides)

Source: stackoverflow.com

Similar codes

1 vote331 views1 answer
Run django app locally
Python
2 votes318 views1 answer
Get all files within multiple directories python
Python
5 votes314 views2 answers
Check if dataframe contains infinity
Python
0 vote338 views1 answer
Sns histplot nan values
Python
0 vote315 views1 answer
Type of tuple in python
Python
0 vote295 views3 answers
Captcha.image install in python
Python
1 vote316 views1 answer
Python print dict new line
Python
4 votes275 views2 answers
Os.walk python
Python
0 vote302 views1 answer
Spotify recommendations
Python
1 vote331 views1 answer
Gimp python
Python