xarray.DataArray.get_axis_num

DataArray.get_axis_num(dim)

Return axis number(s) corresponding to dimension(s) in this array.

Parameters:

dim (str or iterable of str) – Dimension name(s) for which to lookup axes.

Returns:

int or tuple of int – Axis number or numbers corresponding to the given dimensions.