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

10 votes186 views6 answers
Merge dicts python
Python
5 votes256 views1 answer
Resize imshow opencv python
Python
0 vote181 views1 answer
Convert string input into a nested tuple in python
Python
0 vote186 views1 answer
Moving averages python
Python
0 vote187 views1 answer
K-means clustering and disabling clusters
Python
1 vote174 views1 answer
Python - How To Ways to Remove xa0 From a String
Python
1 vote165 views1 answer
How to scroll down to end of page in selenium python
Python
0 vote183 views1 answer
How to import qpalette pyqt5
Python
1 vote184 views1 answer
How to get the user ip in djagno
Python
0 vote187 views1 answer
Forward fill pandas ffill
Python