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

15 votes141 views5 answers
Mouse in pygame
Python
0 vote140 views1 answer
Comma separator f string python
Python
3 votes146 views3 answers
How to take matrix input in python
Python
0 vote141 views1 answer
Gets an existing SparkSession or, if there is no existing one, creates a new
Python
0 vote156 views1 answer
Fetch row where column is missing pandas
Python
0 vote146 views1 answer
Degree symbol in python
Python
-1 vote146 views1 answer
Pseudocode examples for beginners
Python
0 vote129 views1 answer
Download viper for python ubutunu
Python
0 vote181 views1 answer
Find all nan columns pandas
Python
0 vote148 views1 answer
How to install ffmpeg python heroku
Python