H3Neighbourhood

srai.neighbourhoods.H3Neighbourhood(
    regions_gdf=None, include_center=False
)

Bases: Neighbourhood[str]

H3 Neighbourhood.

This class allows to get the neighbours of an H3 region.

If a regions GeoDataFrame is provided, only the neighbours that are in the regions GeoDataFrame will be returned by the methods of this instance. NOTICE: If a region is a part of the k-th ring of a region and is included in the GeoDataFrame, it will be returned by get_neighbours_at_distance method with distance k even when there is no path of length k between the two regions.

PARAMETER	DESCRIPTION
regions_gdf	The regions that are being analyzed. The H3Neighbourhood will only look for neighbours among these regions. Defaults to None. TYPE: Optional[GeoDataFrame] DEFAULT: None
include_center	Whether to include the region itself in the neighbours. TYPE: bool DEFAULT: False

Source code in srai/neighbourhoods/h3_neighbourhood.py

def __init__(
    self, regions_gdf: Optional[gpd.GeoDataFrame] = None, include_center: bool = False
) -> None:
    """
    Initializes the H3Neighbourhood.

    If a regions GeoDataFrame is provided, only the neighbours
    that are in the regions GeoDataFrame will be returned by the methods of this instance.
    NOTICE: If a region is a part of the k-th ring of a region
        and is included in the GeoDataFrame, it will be returned
        by get_neighbours_at_distance method with distance k
        even when there is no path of length k between the two regions.

    Args:
        regions_gdf (Optional[gpd.GeoDataFrame], optional): The regions that are being analyzed.
            The H3Neighbourhood will only look for neighbours among these regions.
            Defaults to None.
        include_center (bool): Whether to include the region itself in the neighbours.
        This is the default value used for all the methods of the class,
        unless overridden in the function call.
    """
    super().__init__(include_center)
    self._available_indices: Optional[set[str]] = None
    if regions_gdf is not None:
        self._available_indices = set(regions_gdf.index)

get_neighbours(index, include_center=None)

Get the direct neighbours of an H3 region using its index.

PARAMETER	DESCRIPTION
index	H3 index of the region. TYPE: str
include_center	Whether to include the region itself in the neighbours. TYPE: Optional[bool] DEFAULT: None

RETURNS	DESCRIPTION
set[str]	Set[str]: Indexes of the neighbours.

Source code in srai/neighbourhoods/h3_neighbourhood.py

def get_neighbours(self, index: str, include_center: Optional[bool] = None) -> set[str]:
    """
    Get the direct neighbours of an H3 region using its index.

    Args:
        index (str): H3 index of the region.
        include_center (Optional[bool]): Whether to include the region itself in the neighbours.
        If None, the value set in __init__ is used. Defaults to None.

    Returns:
        Set[str]: Indexes of the neighbours.
    """
    return self.get_neighbours_up_to_distance(index, 1, include_center)

get_neighbours_up_to_distance(
    index, distance, include_center=None, unchecked=False
)

Get the neighbours of an H3 region up to a certain distance.

PARAMETER	DESCRIPTION
index	H3 index of the region. TYPE: str
distance	Distance to the neighbours. TYPE: int
include_center	Whether to include the region itself in the neighbours. If None, the value set in init is used. Defaults to None. TYPE: Optional[bool] DEFAULT: None
unchecked	Whether to check if the neighbours are in the available indices. TYPE: bool DEFAULT: False

RETURNS	DESCRIPTION
set[str]	Set[str]: Indexes of the neighbours up to the given distance.

Source code in srai/neighbourhoods/h3_neighbourhood.py

def get_neighbours_up_to_distance(
    self,
    index: str,
    distance: int,
    include_center: Optional[bool] = None,
    unchecked: bool = False,
) -> set[str]:
    """
    Get the neighbours of an H3 region up to a certain distance.

    Args:
        index (str): H3 index of the region.
        distance (int): Distance to the neighbours.
        include_center (Optional[bool]): Whether to include the region itself in the neighbours.
            If None, the value set in __init__ is used. Defaults to None.
        unchecked (bool): Whether to check if the neighbours are in the available indices.

    Returns:
        Set[str]: Indexes of the neighbours up to the given distance.
    """
    if self._distance_incorrect(distance):
        return set()

    neighbours: set[str] = set(h3.grid_disk(index, distance))
    neighbours = self._handle_center(
        index, distance, neighbours, at_distance=False, include_center_override=include_center
    )
    if unchecked:
        return neighbours
    return self._select_available(neighbours)

get_neighbours_at_distance(
    index, distance, include_center=None
)

Get the neighbours of an H3 region at a certain distance.

PARAMETER	DESCRIPTION
index	H3 index of the region. TYPE: str
distance	Distance to the neighbours. TYPE: int
include_center	Whether to include the region itself in the neighbours. TYPE: Optional[bool] DEFAULT: None

RETURNS	DESCRIPTION
set[str]	Set[str]: Indexes of the neighbours at the given distance.

Source code in srai/neighbourhoods/h3_neighbourhood.py

def get_neighbours_at_distance(
    self, index: str, distance: int, include_center: Optional[bool] = None
) -> set[str]:
    """
    Get the neighbours of an H3 region at a certain distance.

    Args:
        index (str): H3 index of the region.
        distance (int): Distance to the neighbours.
        include_center (Optional[bool]): Whether to include the region itself in the neighbours.
        If None, the value set in __init__ is used. Defaults to None.

    Returns:
        Set[str]: Indexes of the neighbours at the given distance.
    """
    if self._distance_incorrect(distance):
        return set()

    neighbours: set[str] = set(h3.grid_ring(index, distance))
    neighbours = self._handle_center(
        index, distance, neighbours, at_distance=True, include_center_override=include_center
    )
    return self._select_available(neighbours)