Add index and columns as properties to Styler

[Rishab260]

• Closes pandas Styler has no .columns attr defined #1102
• Tests added: test_styler_columns_and_index in tests/test_styler.py.
• Tested with pytest tests/test_styler.py and mypy tests/test_styler.py

Changes As Per Discussion

• Addition of index and columns to Styler in style.pyi
• Addition of tests in tests/test_style.py for testing index and columns

Related Issues and PRs

• DOC: Missing documentation for Styler.columns and Styler.index pandas#60815
• DOC: Add missing attributes to Styler class pandas#60979

Also See [Source: link ]

from typing_extensions import reveal_type
import pandas as pd
from pandas.io.formats import style as pd_style


def f(s: pd_style.Styler) -> pd_style.Styler:
    print(s.columns)
    fmt = "{:.2%}"
    return s.format(fmt)


df = pd.DataFrame({"a": [1, 2, 3]})
df2 = df.style.pipe(f)
reveal_type(df2)

Output:

$ mypy pandas-stubs/mytest/t.py
pandas-stubs/mytest/t.py:14: note: Revealed type is "pandas.io.formats.style.Styler"
Success: no issues found in 1 source file

Additional Comments

First PR on pandas-stubs, your feedback will be highly valuable to me, Thanks!

[Rishab260]

/pandas_nightly

[Dr-Irv]

You have issues with some formatting and ordering of imports.

You should set up the pre-commit so things are properly formatted. See the output here:
https://github.com/pandas-dev/pandas-stubs/actions/runs/13605234204/job/38040314429?pr=1135

And instructions here for setting up your local testing environment:
https://github.com/pandas-dev/pandas-stubs/blob/main/docs/setup.md

[Rishab260]

You have issues with some formatting and ordering of imports.

You should set up the pre-commit so things are properly formatted. See the output here:
https://github.com/pandas-dev/pandas-stubs/actions/runs/13605234204/job/38040314429?pr=1135

And instructions here for setting up your local testing environment:
https://github.com/pandas-dev/pandas-stubs/blob/main/docs/setup.md

Sure, I will make the changes, I actually had some trouble setting up the development environment following those instructions. For some reason unless I disable the KMP_AFFINITY pytest just crashes.

[Rishab260]

/pandas_nightly

[Rishab260]

You have issues with some formatting and ordering of imports.

You should set up the pre-commit so things are properly formatted. See the output here: https://github.com/pandas-dev/pandas-stubs/actions/runs/13605234204/job/38040314429?pr=1135

And instructions here for setting up your local testing environment: https://github.com/pandas-dev/pandas-stubs/blob/main/docs/setup.md

Hi @Dr-Irv, This is now completed, PTAL. Thanks for the instructions and feedback.

[Dr-Irv]

thanks @Rishab260