# 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