you didn't provide any hints why it wouldn't work for you, so i recommend you look at scikit-learn's neighbors module (i.e., construct NearestNeighbors object and query it).

https://scikit-learn.org/stable/modules/neighbors.html#unsup...