コード例 #1
0
def create_index(data, leafsize=10):
    """
    Creates the k-D tree index.

    :param data: list of HyperPoints to index
    """
    spatial_points = data[['latitude', 'longitude']]
    if hasattr(data, 'data'):
        mask = np.ma.getmask(data.data).ravel()
    else:
        mask = None
    return HaversineDistanceKDTree(spatial_points,
                                   mask=mask,
                                   leafsize=leafsize)