original_array = np.array([0,1,2,3,4])

np.array([array for array in zip(original_array[:-2], original_array[1:], original_array[2:])])

じゃダメなん？