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