Numpy rolling 2d

def rolling_window(a, window_size):
    shape = (a.shape[0] - window_size + 1, window_size) + a.shape[1:]
    strides = (a.strides[0],) + a.strides
    return np.lib.stride_tricks.as_strided(a, shape=shape, strides=strides)

Source: stackoverflow.com

Similar codes

2 votes168 views1 answer
How to open file dialog in pytohn
Python
3 votes147 views1 answer
Extended euclidean python
Python
9 votes134 views2 answers
Django import Q
Python
0 vote143 views1 answer
Django email change sender name
Python
0 vote151 views1 answer
Pattern program in python A aB bCc DdEe
Python
1 vote137 views1 answer
Get website content with beautifulsoup
Python
0 vote140 views1 answer
Try for loop python
Python
0 vote229 views1 answer
Pip install contractions
Python
0 vote155 views1 answer
Find highest correlation pairs pandas
Python
0 vote153 views2 answers
Write a program to check whether a character is vowel or consonant in python
Python