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 vote189 views1 answer
Python roll a die
Python
4 votes192 views2 answers
Pandas to_csv delimiter
Python
0 vote205 views1 answer
Save dictionary to file numpy
Python
0 vote193 views1 answer
Transformer un dictionnaire en liste python
Python
0 vote192 views1 answer
Python string: immutable string
Python
0 vote197 views1 answer
Python open multiple .py windows
Python
2 votes427 views1 answer
Python elementtree build xml
Python
2 votes195 views2 answers
Python sort file names with numbers
Python
0 vote213 views1 answer
Identity matrix in python
Python
34 votes174 views6 answers
Class python example
Python