Function latlon_coords

src/wrf/util.py:3422–3464

Return the latitude and longitude coordinates from a :class:`xarray.DataArray` object. Args: var (:class:`xarray.DataArray`): A variable. as_np (:obj:`bool`): Set to True to return the coordinates as :class:`numpy.ndarray` objects instead of :class

(var, as_np=False)

def latlon_coords(var, as_np=False):
    """Return the latitude and longitude coordinates from a
    :class:`xarray.DataArray` object.

    Args:

        var (:class:`xarray.DataArray`):  A variable.

        as_np (:obj:`bool`): Set to True to return the coordinates as
            :class:`numpy.ndarray` objects instead of
            :class:`xarray.DataArray` objects.

    Returns:

        :obj:`tuple`: The latitude and longitude coordinate variables.

    """

    if not xarray_enabled():
        raise ValueError("xarray is not installed or is disabled")

    try:
        var_coords = var.coords
    except AttributeError:
        raise ValueError("'var' object does not contain coordinate "
                         "attributes")

    latname, lonname, _ = _find_coord_names(var_coords)
    try:
        lats = var_coords[latname]
    except KeyError:
        raise ValueError("'var' object does not contain a latitude "
                         "coordinate")
    try:
        lons = var_coords[lonname]
    except KeyError:
        raise ValueError("'var' object does not contain a longitude "
                         "coordinate")

    if as_np:
        return to_np(lats), to_np(lons)

    return lats, lons


def get_cartopy(var=None, wrfin=None, varname=None, timeidx=0, method="cat",