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

11 votes294 views3 answers
How to get a specific character in a string on number python
Python
0 vote273 views1 answer
Python string: indexing and slicing string
Python
0 vote309 views1 answer
Initial TypedMultipleChoiceField django
Python Django
0 vote279 views1 answer
Python numpy read from stdin
Python
0 vote283 views1 answer
Branchless if python
Python
0 vote250 views1 answer
Pyserial read
Python
0 vote258 views1 answer
Dice throw program in python
Python
1 vote268 views1 answer
Round godot
Python
1 vote257 views3 answers
Isprime python
Python
0 vote273 views2 answers
Python open google
Python