Skip to content

DataFrame.to_html() #387

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

Closed
wants to merge 1 commit into from
Closed

DataFrame.to_html() #387

Show file tree
Hide file tree
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
146 changes: 138 additions & 8 deletions pandas/core/frame.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -678,8 +678,9 @@ def to_csv(self, path, sep=",", na_rep='', cols=None, header=True,
def to_string(self, buf=None, columns=None, colSpace=None,
na_rep='NaN', formatters=None, float_format=None,
sparsify=True, nanRep=None, index_names=True):


"""
Render a DataFrame to a console-friendly tabular output.
"""
if nanRep is not None: # pragma: no cover
import warnings
warnings.warn("nanRep is deprecated, use na_rep",
Expand All @@ -693,6 +694,24 @@ def to_string(self, buf=None, columns=None, colSpace=None,
float_format=float_format,
sparsify=sparsify,
index_names=index_names)
formatter.to_string()

if buf is None:
return formatter.buf.getvalue()

def to_html(self, buf=None, columns=None, colSpace=None,
na_rep='NaN', formatters=None, float_format=None,
sparsify=True, index_names=True):
"""
Render a DataFrame to a html table.
"""
formatter = _DataFrameFormatter(self, buf=buf, columns=columns,
col_space=colSpace, na_rep=na_rep,
formatters=formatters,
float_format=float_format,
sparsify=sparsify,
index_names=index_names)
formatter.to_html()

if buf is None:
return formatter.buf.getvalue()
Expand Down Expand Up @@ -3230,7 +3249,10 @@ def combineMult(self, other):

class _DataFrameFormatter(object):
"""
Render a console-friendly tabular output of a DataFrame
Render a DataFrame

self.to_string() : console-friendly tabular output
self.to_html() : html table
"""
def __init__(self, frame, buf=None, columns=None, col_space=None,
na_rep='NaN', formatters=None, float_format=None,
Expand All @@ -3250,9 +3272,10 @@ def __init__(self, frame, buf=None, columns=None, col_space=None,
else:
self.columns = frame.columns

self._write_to_buffer()

def _write_to_buffer(self):
def to_string(self):
"""
Render a DataFrame to a console-friendly tabular output.
"""
frame = self.frame
format_col = self._get_column_formatter()

Expand Down Expand Up @@ -3280,6 +3303,110 @@ def _write_to_buffer(self):

self.buf.writelines(to_write)

def to_html(self):
"""
Render a DataFrame to a html table.
"""
def write(buf, s, indent=0):
buf.write(unicode((' ' * indent) + str(s) + '\n'))

def write_th(buf, s, indent=0):
write(buf, '<th>%s</th>' % str(s), indent)

def write_td(buf, s, indent=0):
write(buf, '<td>%s</td>' % str(s), indent)

def write_tr(buf, l, indent=0, indent_delta=4, header=False):
write(buf, '<tr>', indent)
indent += indent_delta
if header:
for s in l:
write_th(buf, s, indent)
else:
for s in l:
write_td(buf, s, indent)
indent -= indent_delta
write(buf, '</tr>', indent)

def single_column_table(column):
table = '<table><tbody>'
for i in column:
table += ('<tr><td>%s</td></tr>' % str(i))
table += '</tbody></table>'
return table

def single_row_table(row):
table = '<table><tbody><tr>'
for i in row:
table += ('<td>%s</td>' % str(i))
table += '</tr></tbody></table>'
return table

indent = 0
indent_delta = 2
frame = self.frame
buf = self.buf
format_col = self._get_column_formatter()

write(buf, '<table border="1">', indent)

if len(frame.columns) == 0 or len(frame.index) == 0:
write(buf, '<tbody>', indent + indent_delta)
write_tr(buf,
[repr(frame.index),
'Empty %s' % type(self.frame).__name__],
indent + (2 * indent_delta),
indent_delta)
write(buf, '</tbody>', indent + indent_delta)
else:
indent += indent_delta
write(buf, '<thead>', indent)
row = []

if isinstance(frame.index, MultiIndex):
if self.has_index_names:
row.extend(frame.index.names)
else:
row.extend([''] * frame.index.nlevels)
else:
row.append(' ')

if isinstance(frame.columns, MultiIndex):
row.extend([single_column_table(c) for c in frame.columns])
if self.has_column_names:
names = single_column_table(frame.columns.names)
idx = len(frame.columns)
row[-idx] = single_row_table([names, row[-idx]])
else:
row.append('')

indent += indent_delta
write_tr(buf,
row,
indent,
indent_delta,
header=True)
write(buf, '</thead>', indent)

write(buf, '<tbody>', indent)
for i in range(len(frame)):
row = []
try:
row.extend(frame.index[i])
except TypeError:
row.append(frame.index[i])
for column in frame.columns:
row.append(format_col(column, i))
write_tr(buf,
row,
indent,
indent_delta)
indent -= indent_delta
write(buf, '</body>', indent)
indent -= indent_delta

write(buf, '</table>', indent)

def _get_column_formatter(self):
from pandas.core.common import _format

Expand All @@ -3296,9 +3423,12 @@ def _myformat(v):

formatters = {} if self.formatters is None else self.formatters

def _format_col(col):
def _format_col(col, i=None):
formatter = formatters.get(col, _myformat)
return [formatter(x) for x in self.frame[col]]
if i == None:
return [formatter(x) for x in self.frame[col]]
else:
return formatter(self.frame[col][i])

return _format_col

Expand Down
28 changes: 28 additions & 0 deletions pandas/tests/test_frame.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1415,6 +1415,34 @@ def test_to_string(self):
frame = DataFrame(index=np.arange(1000))
frame.to_string()

def test_to_html(self):
# big mixed
biggie = DataFrame({'A' : randn(1000),
'B' : tm.makeStringIndex(1000)},
index=range(1000))

biggie['A'][:20] = nan
biggie['B'][:20] = nan
s = biggie.to_html()

buf = StringIO()
retval = biggie.to_html(buf=buf)
self.assert_(retval is None)
self.assertEqual(buf.getvalue(), s)

self.assert_(isinstance(s, basestring))

biggie.to_html(columns=['B', 'A'], colSpace=17)
biggie.to_html(columns=['B', 'A'],
formatters={'A' : lambda x: '%.1f' % x})

biggie.to_html(columns=['B', 'A'], float_format=str)
biggie.to_html(columns=['B', 'A'], colSpace=12,
float_format=str)

frame = DataFrame(index=np.arange(1000))
frame.to_html()

def test_insert(self):
df = DataFrame(np.random.randn(5, 3), index=np.arange(5),
columns=['c', 'b', 'a'])
Expand Down