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 vote152 views1 answer
Python check if number is complex
Python
3 votes145 views1 answer
How to add the column to the beginning of dataframe
Python
0 vote140 views1 answer
Disable sns plot python
Python
0 vote144 views1 answer
To text pandas
Python
0 vote152 views1 answer
Find position of key in dictionary python
Python
0 vote132 views3 answers
Calling a function in python upon entry content changing tkinter
Python
-2 votes138 views1 answer
En_core_web_sm
Python
94 votes142 views4 answers
Json file to dict python
Python
0 vote137 views1 answer
Replace word in column pandas lambda
Python
0 vote153 views1 answer
Pandas isin another dataframe column
Python