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 vote168 views1 answer
Fill in the empty function so that it returns the sum of all the divisors of a number, without including it. A divisor is a number that divides into another without a remainder.
Python
0 vote113 views10 answers
Python
Python
7 votes125 views2 answers
Check package version python
Python
1 vote108 views1 answer
Python sort a list by a custom order
Python
3 votes200 views4 answers
Fraction in python
Python
0 vote109 views1 answer
MIn-Max problem in python
Python
0 vote101 views1 answer
Python ssl module is not available
Python
20 votes112 views6 answers
Sort list in python
Python
0 vote121 views1 answer
How to make an app like word in python
Python
0 vote112 views1 answer
Cos2x
Python