[ENH]: Using `IndexError` instead of `ValueError` at appropriate places?

☑️ I understand it is strictly prohibited to use AI to write issues.

Following the suggestion, functions like expand_dims raise ValueError when the axis is invalid, but it makes more sense for them to raise IndexError instead.