Return axis number(s) corresponding to dimension(s) in this array.
dim (str or iterable of str) – Dimension name(s) for which to lookup axes.
Axis number or numbers corresponding to the given dimensions.
int or tuple of int