why not just use np.delete?