Indeed, came here to say exactly this. K-d trees are even built in to SciPy, which the author probably already has installed (Scikit-Learn also has an implemen

Indeed, came here to say exactly this. K-d trees are even built in to SciPy, which the author probably already has installed (Scikit-Learn also has an implementation).