10 lines
304 B
Python
10 lines
304 B
Python
|
|
def normalize_axis_index(axis, ndim):
|
|
# Check if `axis` is in the correct range and normalize it
|
|
if axis < -ndim or axis >= ndim:
|
|
msg = f"axis {axis} is out of bounds for array of dimension {ndim}"
|
|
raise Exception(msg)
|
|
|
|
if axis < 0:
|
|
axis = axis + ndim
|
|
return axis |