Questions asking us torecommend or find a tool, library or favorite off-site resource are off-topic for Stack Overflow as they tend to attract opinionated answers and spam. Instead,describe the problem and what has been done so far to solve it.
Closed11 years ago.
I would like to calculate K-nearest neighbour in python. what library should i use?
4 Answers4
I think that you should usescikit ann.
There is a good tutorial about the nearest neightbourhere.
According to the documentation :
ann is a SWIG-generated python wrapper for the Approximate Nearest Neighbor (ANN) Library (http://www.cs.umd.edu/~mount/ANN/), developed by David M. Mount and Sunil Arya. ann provides an immutable kdtree implementation (via ANN) which can perform k-nearest neighbor and approximate k
4 Comments
Here is a script comparing scipy.spatial.cKDTree and pyflann.FLANN. See for yourself which one is faster for your application.
import cProfileimport numpy as npimport osimport pyflannimport scipy.spatial# Config paramsdim = 4data_size = 1000test_size = 1# Generate datanp.random.seed(1)dataset = np.random.rand(data_size, dim)testset = np.random.rand(test_size, dim)def test_pyflann_flann(num_reps): flann = pyflann.FLANN() for rep in range(num_reps): params = flann.build_index(dataset, target_precision=0.0, log_level='info') result = flann.nn_index(testset, 5, checks=params['checks'])def test_scipy_spatial_kdtree(num_reps): flann = pyflann.FLANN() for rep in range(num_reps): kdtree = scipy.spatial.cKDTree(dataset, leafsize=10) result = kdtree.query(testset, 5)num_reps = 1000cProfile.run('test_pyflann_flann(num_reps); test_scipy_spatial_kdtree(num_reps)', 'out.prof')os.system('runsnake out.prof')Comments
scipy.spatial.cKDTreeis fast and solid.For an example of using it for NN interpolation, see (ahem)inverse-distance-weighted-idw-interpolation-with-python on SO.
(If you could say e.g. "I have 1M points in 3d, and want k=5 nearest neighbors of 1k new points",you might get better answers or code examples.
What do you want to do with the neighbors once you've found them ?)
Comments
It is natively in scipy if you're looking to do a kd-tree approach:http://docs.scipy.org/doc/scipy/reference/generated/scipy.spatial.KDTree.html#scipy.spatial.KDTree
Comments
Explore related questions
See similar questions with these tags.