add unique to python

h2oai · Jul 17, 2015 · 90423fa · 90423fa
1 parent 713b27f
commit 90423fa
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/h2o-py/h2o/frame.py b/h2o-py/h2o/frame.py
@@ -271,7 +271,7 @@ def filterNACols(self, frac=0.2):
     :param frac: Fraction of NAs in the column.
     :return: A  list of column indices.
     """
-    return H2OFrame(ExprNode("filterNACols"), self, frac)._frame()
+    return H2OFrame(expr=ExprNode("filterNACols", self, frac))._frame()
 
   def dim(self):
     """
@@ -281,6 +281,14 @@ def dim(self):
     """
     return [self.nrow(), self.ncol()]
 
+  def unique(self):
+    """
+    Extract the unique values in the column.
+
+    :return: A new H2OFrame of just the unique values in the column.
+    """
+    return H2OFrame(expr=ExprNode("unique", self))._frame()
+
   def show(self): self.head(10,sys.maxint)  # all columns
 
   def head(self, rows=10, cols=200, **kwargs):

diff --git a/h2o-py/h2o/h2o.py b/h2o-py/h2o/h2o.py
@@ -187,9 +187,6 @@ def get_model(model_id):
   else:                           raise NotImplementedError(model_type)
 
 
-def str(frame):
-  pass
-
 def get_frame(frame_id):
   """
   Obtain a handle to the frame in H2O with the frame_id key.