[ZEPPELIN-1255] Add cast to string in z.show() for Pandas DataFrame

This commit is contained in:
paulbustios 2016-07-30 02:14:08 -03:00
parent 848dbd0302
commit 4a8c0a9e33

View file

@ -140,19 +140,19 @@ class PyZeppelinContext(object):
"""
limit = len(df) > self.max_result
header_buf = io.StringIO("")
header_buf.write(df.columns[0])
header_buf.write(str(df.columns[0]))
for col in df.columns[1:]:
header_buf.write("\t")
header_buf.write(col)
header_buf.write(str(col))
header_buf.write("\n")
body_buf = io.StringIO("")
rows = df.head(self.max_result).values if limit else df.values
for row in rows:
body_buf.write(row[0])
body_buf.write(str(row[0]))
for cell in row[1:]:
body_buf.write("\t")
body_buf.write(cell)
body_buf.write(str(cell))
body_buf.write("\n")
body_buf.seek(0); header_buf.seek(0)
#TODO(bzz): fix it, so it shows red notice, as in Spark