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 vote263 views1 answer
Rolling std dev of a pandas series
Python
0 vote262 views1 answer
Validate string using six library python
Python
2 votes245 views3 answers
Display entire row pandas
Python
4 votes264 views1 answer
Print a random word from list python
Python
-2 votes255 views1 answer
Youtube dl python
Python
5 votes313 views2 answers
Copy directory from one location to another python
Python
0 vote276 views1 answer
List of lists to table python
Python
33 votes263 views3 answers
Add dataframe column to set
Python
5 votes250 views2 answers
If df[col].unique()==2
Python
25 votes223 views2 answers
Check if a list of strings is in a string python
Python