fix SQL bucket group, orderby op code

yennanliu · Dec 27, 2018 · aef72cd · aef72cd
1 parent db70ded
commit aef72cd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/spark_/spark_SQL_analysis.java b/spark_/spark_SQL_analysis.java
@@ -56,13 +56,13 @@ public static void main(String[] args) throws Exception {
 
 
         Dataset<Row> responseWithSalaryBucket = castedResponse.withColumn(
-                PICKUP_LATITUDE_BUCKET, col(PICKUP_LATITUDE).divide(2).cast("integer").multiply(3));
+                PICKUP_LATITUDE_BUCKET, col(PICKUP_LATITUDE).divide(0.2).cast("integer").multiply(0.2));
 
-        System.out.println("=== With salary bucket column ===");
+        System.out.println("=== With PICKUP_LATITUDE_BUCKET column ===");
         responseWithSalaryBucket.select(col(PICKUP_LATITUDE), col(PICKUP_LATITUDE_BUCKET)).show();
 
-        //System.out.println("=== Group by salary bucket ===");
-        //responseWithSalaryBucket.groupBy(SALARY_MIDPOINT_BUCKET).count().orderBy(col(SALARY_MIDPOINT_BUCKET)).show();
+        System.out.println("=== Group by PICKUP_LATITUDE_BUCKET ===");
+        responseWithSalaryBucket.groupBy(PICKUP_LATITUDE_BUCKET).count().orderBy(col(PICKUP_LATITUDE_BUCKET)).show();
 
         session.stop();
     }