ENH: use partial sort for kneighbors selection