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

0 vote358 views2 answers
Numpy roundup to nearest 5
Python
2 votes326 views1 answer
Calendar program in python
Python
0 vote404 views1 answer
Mongodb get first 10 records
Python
0 vote367 views1 answer
Reference other libraries in library
Python
0 vote337 views2 answers
How to terminate subprocess.call in python
Python
0 vote344 views1 answer
Python pick one item from list
Python
0 vote383 views1 answer
Python string lower method
Python
0 vote340 views1 answer
Python skip line
Python
0 vote334 views1 answer
2D array questions python
Python
1 vote362 views1 answer
Django login redirect
Python Django