# Import the NumPy library import numpy as np def rolling(a, window): return np.lib.stride_tricks.sliding_window_view(a, window) Z = rolling(np.arange(10), 3) print(Z) # printing results