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

[attack68]

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 REF: axis in [...] becomes _get_axis_number(axis) in Styler #43078
• test_label_or_level_utils.py REF: Replacing axis in with _get_axis_number(axis) #43131
• REF: axis in [...] becomes _get_axis_number(axis) in Styler #43121

[rhshadrach]

I think this applies to all code, not just the style, is that right? Or would you like to keep this issue focused on just the styler?

[attack68]

@rhshadrach I don't know. I know that it applies to much of the stuff I wrote in Styler that @jreback pointed out to me. If it applies more broadly can convert this issue into a master tracker and try to locate more examples..

[jreback]

yeah this was mainly for Styler, but its possible we do this elsewhere in the codebase so should clean those up (and ideally have a pre-commit rule but might be non-trivial)

[dhivyadharshin]

Can I work on this issues?