diff --git a/common/client/src/main/java/zingg/common/client/ZFrame.java b/common/client/src/main/java/zingg/common/client/ZFrame.java index 8b9520321..7c1c22435 100644 --- a/common/client/src/main/java/zingg/common/client/ZFrame.java +++ b/common/client/src/main/java/zingg/common/client/ZFrame.java @@ -16,8 +16,7 @@ public interface ZFrame { public String[] columns(); public ZFrame select(String... string); public ZFrame select(List cols); - public ZFrame select(String col, String... cols); - public ZFrame select(String col); + public ZFrame select(C col); public ZFrame selectExpr(String... col); public ZFrame distinct(); public List collectAsList(); diff --git a/spark/client/src/main/java/zingg/spark/client/SparkFrame.java b/spark/client/src/main/java/zingg/spark/client/SparkFrame.java index 7b1bc13cc..17ea47bf7 100644 --- a/spark/client/src/main/java/zingg/spark/client/SparkFrame.java +++ b/spark/client/src/main/java/zingg/spark/client/SparkFrame.java @@ -44,14 +44,18 @@ public ZFrame, Row, Column> select(Column... cols) { return new SparkFrame(df.select(cols)); } + public ZFrame, Row, Column> select(Column col) { + return new SparkFrame(df.select(col)); + } + public ZFrame, Row, Column> select(List cols){ return new SparkFrame(df.select(JavaConverters.asScalaIteratorConverter(cols.iterator()).asScala().toSeq())); } - public ZFrame, Row, Column> select(String col) { - return new SparkFrame(df.select(col)); + public ZFrame, Row, Column> select(String... col) { + return toDF(col); } public ZFrame, Row, Column> selectExpr(String... col) {