endas.localization.SpatialQuery

class endas.localization.SpatialQuery

Bases: object

Lookup of observations or state vector elements by location.

ndim

Returns the number of spatial dimensions. This is a convenience shortcut to self.cs.ndim.

cs

Returns the coordinate system the query operates in.

range_query(coord, range, distances=False)

Looks up items up to the given distance from the given coordinate.

Parameters:
  • coord – Array of length self.ndim representing the queried coordinate.
  • range – Search range. All items within this distance from coord will be returned.
  • distances – If True, distances of the returned elements from the query coordinate will also be returned.
Returns:

Array of indexes of items that are within the range. If distances=True was passed, a second array of the same length containing distances of the corresponding items is also returned.