Finding index of nearest point in numpy arrays of x and y coordinates

by

Here is a scipy.spatial.KDTree example

In [1]: from scipy import spatial

In [2]: import numpy as np

In [3]: A = np.random.random((10,2))*100

In [4]: A
Out[4]:
array([[ 68.83402637,  38.07632221],
       [ 76.84704074,  24.9395109 ],
       [ 16.26715795,  98.52763827],
       [ 70.99411985,  67.31740151],
       [ 71.72452181,  24.13516764],
       [ 17.22707611,  20.65425362],
       [ 43.85122458,  21.50624882],
       [ 76.71987125,  44.95031274],
       [ 63.77341073,  78.87417774],
       [  8.45828909,  30.18426696]])

In [5]: pt = [6, 30]  # <-- the point to find

In [6]: A[spatial.KDTree(A).query(pt)[1]] # <-- the nearest point 
Out[6]: array([  8.45828909,  30.18426696])

#how it works!
In [7]: distance,index = spatial.KDTree(A).query(pt)

In [8]: distance # <-- The distances to the nearest neighbors
Out[8]: 2.4651855048258393

In [9]: index # <-- The locations of the neighbors
Out[9]: 9

#then 
In [10]: A[index]
Out[10]: array([  8.45828909,  30.18426696])

More Related Contents:

  1. Find elements of array one nearest to elements of array two
  2. How to invert a permutation array in numpy
  3. Compute *rolling* maximum drawdown of pandas Series
  4. NumPy grouping using itertools.groupby performance
  5. Sieve of Eratosthenes – Primes between X and N
  6. What is the Big O Complexity of Reversing the Order of Columns in Pandas DataFrame?
  7. Pandas conditional creation of a series/dataframe column
  8. How to convert a PIL Image into a numpy array?
  9. Roll rows of a matrix independently
  10. How did Python implement the built-in function pow()?
  11. Pandas Timedelta in Days
  12. Vectorized searchsorted numpy
  13. How to remove blanks/NA’s from dataframe and shift the values up
  14. Why do “Not a Number” values equal True when cast as boolean in Python/Numpy?
  15. NumPy: function for simultaneous max() and min()
  16. When splitting an empty string in Python, why does split() return an empty list while split(‘\n’) returns [”]?
  17. Using Numpy Vectorize on Functions that Return Vectors
  18. Convert numpy.datetime64 to string object in python
  19. Matplotlib: how to make imshow read x,y coordinates from other numpy arrays?
  20. Numpy individual element access slower than for lists
  21. Why using an array as an index changes the shape of a multidimensional ndarray?
  22. How to get stable results with TensorFlow, setting random seed
  23. How to multiply individual elements of a list with a number?
  24. Installing lapack for numpy
  25. Adding colors to a 3d quiver plot in matplotlib
  26. Python: powerset of a given set with generators [duplicate]
  27. sampling random floats on a range in numpy
  28. How does the axis parameter from NumPy work?
  29. Fast interpolation of grid data
  30. numpy faster than numba and cython , how to improve numba code

Leave a Comment