def shift(arr, steps, val=0): res_arr = np.roll(arr, steps) np.put(res_arr, range(steps), val) return res_arr