GriSPy: A Python package for fixed-radius nearest neighbors search
Abstract
We present a new regular grid search algorithm for quick fixed-radius nearest-neighbor lookup developed in Python. This module indexes a set of k-dimensional points in a regular grid, with optional periodic conditions, providing a fast approach for nearest neighbors queries. In this first installment, we provide three types of queries: bubble, shell and the nth-nearest. For these queries we include three different metrics of interest in astronomy, namely, the euclidean, the haversine and the Vincenty, the last two in spherical coordinates. We also provide the possibility of using a custom distance function. This package results particularly useful for large datasets where a brute-force search turns impractical.
- Publication:
-
Astronomy and Computing
- Pub Date:
- January 2021
- DOI:
- 10.1016/j.ascom.2020.100443
- arXiv:
- arXiv:1912.09585
- Bibcode:
- 2021A&C....3400443C
- Keywords:
-
- Data mining;
- Nearest-neighbor search;
- Methods;
- Data analysis;
- Astroinformatics;
- Python package;
- Astrophysics - Instrumentation and Methods for Astrophysics
- E-Print:
- Accepted for publication in Astronomy and Computing on December 16, 2020. 12 pages, 8 figures