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 vote146 views1 answer
How to implement dfa in python
Python
0 vote168 views2 answers
Convert pdf folder to excell pandas
Python
23 votes310 views3 answers
Install django rest_framework
Python Django
0 vote148 views1 answer
R Return each result with an index
Python
6 votes172 views3 answers
Cosine Similarity numpy
Python
0 vote146 views1 answer
Swap two lists without using third variable python
Python
1 vote196 views1 answer
Check if queryset is empty django template
Python Django
0 vote174 views1 answer
How to delete all instances of a model in django
Python Django
8 votes168 views1 answer
How to run python in atom
Python
0 vote134 views1 answer
Get mode dataframe
Python