[groupby.js][s]: obtain groupby column dtype to fix #183

Author: steveoni

danfojs-node/dist/core/groupby.js

@@ -14,12 +14,13 @@ var _series = require("./series");
 const utils = new _utils.Utils();
 
 class GroupBy {
-  constructor(col_dict, key_col, data, column_name) {
+  constructor(col_dict, key_col, data, column_name, col_dtype) {
     this.key_col = key_col;
     this.col_dict = col_dict;
     this.data = data;
     this.column_name = column_name;
     this.data_tensors = {};
+    this.col_dtype = col_dtype;
   }
 
   group() {
@@ -118,7 +119,7 @@ class GroupBy {
       }
     }
 
-    const gp = new GroupBy(null, this.key_col, null, col_names);
+    const gp = new GroupBy(null, this.key_col, null, col_names, this.col_dtype);
     gp.group_col = group_col;
     gp.group_col_name = col_names;
     return gp;
@@ -315,8 +316,8 @@ class GroupBy {
               for (let j = 0; j < col_data.length; j++) {
                 if (typeof key_data[j] === "undefined") {
                   key_data[j] = [];
-                  key_data[j][0] = isNaN(parseInt(key_1)) ? key_1 : parseInt(key_1);
-                  key_data[j][1] = isNaN(parseInt(key_2)) ? key_2 : parseInt(key_2);
+                  key_data[j][0] = this.col_dtype[0] === "string" ? key_1 : parseInt(key_1);
+                  key_data[j][1] = this.col_dtype[1] === "string" ? key_2 : parseInt(key_2);
                   key_data[j].push(col_data[j]);
                 } else {
                   key_data[j].push(col_data[j]);
@@ -326,8 +327,8 @@ class GroupBy {
 
             df_data.push(...key_data);
           } else {
-            key_data[0] = isNaN(parseInt(key_1)) ? key_1 : parseInt(key_1);
-            key_data[1] = isNaN(parseInt(key_2)) ? key_2 : parseInt(key_2);
+            key_data[0] = this.col_dtype[0] === "string" ? key_1 : parseInt(key_1);
+            key_data[1] = this.col_dtype[1] === "string" ? key_2 : parseInt(key_2);
             key_data.push(...k_data);
             df_data.push(key_data);
           }
@@ -360,7 +361,7 @@ class GroupBy {
             for (let j = 0; j < col_data.length; j++) {
               if (typeof key_data[j] === "undefined") {
                 key_data[j] = [];
-                key_data[j][0] = isNaN(parseInt(key_1)) ? key_1 : parseInt(key_1);
+                key_data[j][0] = this.col_dtype[0] === "string" ? key_1 : parseInt(key_1);
                 key_data[j].push(col_data[j]);
               } else {
                 key_data[j].push(col_data[j]);
@@ -370,7 +371,7 @@ class GroupBy {
 
           df_data.push(...key_data);
         } else {
-          key_data[0] = isNaN(parseInt(key_1)) ? key_1 : parseInt(key_1);
+          key_data[0] = this.col_dtype[0] === "string" ? key_1 : parseInt(key_1);
           key_data.push(...key_val);
           df_data.push(key_data);
         }