def impute_zeros(x, y): imputed_y = np.zeros(np.ptp(x) + 1) np.put(imputed_y, ind=np.array(x) - min(x), v=np.array(y), mode='clip') return imputed_y.tolist()