Skip to content

⚡️ Speed up method Styler.set_table_attributes by 8% #409

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
codeflash-ai wants to merge 1 commit into
base: main
Choose a base branch
from
Open

⚡️ Speed up method Styler.set_table_attributes by 8% #409

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 35 additions & 12 deletions pandas/io/formats/style.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -264,6 +264,34 @@ def __init__(
escape: str | None = None,
formatter: ExtFormatter | None = None,
) -> None:
# Pre-compute get_option values once, avoiding repeated global lookups
_styler_format_opts = {
"thousands": thousands,
"decimal": decimal,
"na_rep": na_rep,
"escape": escape,
"formatter": formatter,
}
opts_missing = [k for k, v in _styler_format_opts.items() if v is None]
if opts_missing:
# If any option is None, fetch all needed at once efficiently
getopt = get_option
thousands = (
thousands
if thousands is not None
else getopt("styler.format.thousands")
)
decimal = (
decimal if decimal is not None else getopt("styler.format.decimal")
)
na_rep = na_rep if na_rep is not None else getopt("styler.format.na_rep")
escape = escape if escape is not None else getopt("styler.format.escape")
formatter = (
formatter
if formatter is not None
else getopt("styler.format.formatter")
)

super().__init__(
data=data,
uuid=uuid,
Expand All @@ -274,13 +302,6 @@ def __init__(
cell_ids=cell_ids,
precision=precision,
)

# validate ordered args
thousands = thousands or get_option("styler.format.thousands")
decimal = decimal or get_option("styler.format.decimal")
na_rep = na_rep or get_option("styler.format.na_rep")
escape = escape or get_option("styler.format.escape")
formatter = formatter or get_option("styler.format.formatter")
# precision is handled by superclass as default for performance

self.format(
Expand Down Expand Up @@ -2483,7 +2504,7 @@ def set_sticky(
for i, level in enumerate(levels_):
styles.append(
{
"selector": f"thead tr:nth-child({level+1}) th",
"selector": f"thead tr:nth-child({level + 1}) th",
"props": props
+ (
f"top:{i * pixel_size}px; height:{pixel_size}px; "
Expand All @@ -2494,7 +2515,7 @@ def set_sticky(
if not all(name is None for name in self.index.names):
styles.append(
{
"selector": f"thead tr:nth-child({obj.nlevels+1}) th",
"selector": f"thead tr:nth-child({obj.nlevels + 1}) th",
"props": props
+ (
f"top:{(len(levels_)) * pixel_size}px; "
Expand All @@ -2514,7 +2535,7 @@ def set_sticky(
styles.extend(
[
{
"selector": f"thead tr th:nth-child({level+1})",
"selector": f"thead tr th:nth-child({level + 1})",
"props": props_ + "z-index:3 !important;",
},
{
Expand Down Expand Up @@ -4109,8 +4130,10 @@ def css_bar(start: float, end: float, color: str) -> str:
if end > start:
cell_css += "background: linear-gradient(90deg,"
if start > 0:
cell_css += f" transparent {start*100:.1f}%, {color} {start*100:.1f}%,"
cell_css += f" {color} {end*100:.1f}%, transparent {end*100:.1f}%)"
cell_css += (
f" transparent {start * 100:.1f}%, {color} {start * 100:.1f}%,"
)
cell_css += f" {color} {end * 100:.1f}%, transparent {end * 100:.1f}%)"
return cell_css

def css_calc(x, left: float, right: float, align: str, color: str | list | tuple):
Expand Down