# Import the NumPy library import numpy as np def rolling(a, window, n): return np.stack([a[i:i+window] for i in range(a.size - window + n)]) Z = rolling(np.arange(10), 3, 1) print(Z) # printing results