xarray.DataArray.get_axis_num

DataArray.get_axis_num(dim)

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

Parameters
dimstr 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.