Skip to content

Commit 651a244

Browse files
committed
refactoring a bit
1 parent 25f4a87 commit 651a244

File tree

3 files changed

+1272
-792
lines changed

3 files changed

+1272
-792
lines changed

pandas/core/format.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -692,7 +692,7 @@ def _format_col(self, i):
692692
space=self.col_space
693693
)
694694

695-
def to_html(self, classes=None):
695+
def to_html(self, classes=None, style=None):
696696
"""
697697
Render a DataFrame to a html table.
698698
"""
@@ -808,9 +808,11 @@ class HTMLFormatter(TableFormatter):
808808

809809
indent_delta = 2
810810

811-
def __init__(self, formatter, classes=None, max_rows=None, max_cols=None):
811+
def __init__(self, formatter, classes=None, max_rows=None, max_cols=None,
812+
style=style):
812813
self.fmt = formatter
813814
self.classes = classes
815+
self.style = style
814816

815817
self.frame = self.fmt.frame
816818
self.columns = self.fmt.tr_frame.columns

pandas/core/frame.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -1349,7 +1349,7 @@ def to_html(self, buf=None, columns=None, col_space=None, colSpace=None,
13491349
header=True, index=True, na_rep='NaN', formatters=None,
13501350
float_format=None, sparsify=None, index_names=True,
13511351
justify=None, bold_rows=True, classes=None, escape=True,
1352-
max_rows=None, max_cols=None, show_dimensions=False, styles=None):
1352+
max_rows=None, max_cols=None, show_dimensions=False, style=None):
13531353
"""
13541354
Render a DataFrame as an HTML table.
13551355
@@ -1388,7 +1388,8 @@ def to_html(self, buf=None, columns=None, col_space=None, colSpace=None,
13881388
max_rows=max_rows,
13891389
max_cols=max_cols,
13901390
show_dimensions=show_dimensions)
1391-
formatter.to_html(classes=classes)
1391+
# TODO: a generic formatter wld b in DataFrameFormatter
1392+
formatter.to_html(classes=classes, style=style)
13921393

13931394
if buf is None:
13941395
return formatter.buf.getvalue()
@@ -1400,6 +1401,7 @@ def style(self, rows=None, columns=None, table=None):
14001401
return self.to_html(styles=styles)
14011402

14021403
def _format_sytle_args(self, rows=None, columns=None, frame=None):
1404+
# TODO: hanlde rows=func.
14031405
rows = {row: [x] if not com.is_list_like(x) else x
14041406
for row, x in rows.items()}
14051407
columns = {col: [x] if not com.is_list_like(x) else x

0 commit comments

Comments
 (0)