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

0 vote290 views1 answer
Get os environment python
Python
2 votes272 views1 answer
Python text fromatting rows
Python
2 votes278 views5 answers
Torch.stack
Python
2 votes293 views4 answers
ModuleNotFoundError: No module named 'tensorflow_io'
Python
5 votes270 views1 answer
Tkiner border
Python
0 vote277 views1 answer
Name decorator in python
Python
23 votes283 views4 answers
Python flat list from list of list
Python
0 vote272 views1 answer
Python time function in for loop
Python
0 vote266 views1 answer
Pandas show head and tail
Python
73 votes265 views6 answers
Remove duplicates python
Python