diff --git a/h2o-py/h2o/frame.py b/h2o-py/h2o/frame.py index 43357489c8a2..79e0bd2ee076 100644 --- a/h2o-py/h2o/frame.py +++ b/h2o-py/h2o/frame.py @@ -307,12 +307,12 @@ def tail(self, rows=10, cols=200, **kwargs): self._eager() nrows = min(self.nrow(), rows) ncols = min(self.ncol(), cols) - colnames = self.names()[0:ncols] start_idx = max(self.nrow()-nrows,0) tail = self[start_idx:(start_idx+nrows),:] res = tail.as_data_frame(False) + colnames = res.pop(0) print "Last {} rows and first {} columns: ".format(nrows,ncols) - h2o.H2ODisplay(res,["Row ID"]+colnames) + h2o.H2ODisplay(res,colnames) return tail def levels(self, col=None):