REF: use `_get_axis_number` instead of `if axis in [...]`:

This PR regards searching the non-test codebase (i.e. don’t find code examples in pandas/tests) for examples like the following:

Try to find and replace logic like:

if axis in [0, "index"]:
    ...
elif axis in [1, "columns"]:
    ...
else:
    ...

with the standard codebase method get_axis_number(axis). This is a DataFrame method so often you need use local DataFrame objects to call this method.

It may also be necessary to change axis args in internal methods to consistently accept the same input format compatible with this method, i.e. axis as an integer (0 or 1).

This is an open issue and contributions are welcome to solve individual files in the code. As an example see #43078 or other merged contributions below.

• style.py #43078
• test_label_or_level_utils.py #43131
• #43121

[attack68]

Anyone is welcome to work on this issue. This will not be solved by just one PR, there are likely many places and many files where the same work needs to be done. Therefore many people can work on this, simultaneously.

[attack68]

Did a quick search of the codebase and outside of all the test files this seems to have been addressed by all contributions now.