update concat to accept both dataframe and series

Author: steveoni

danfojs/src/core/concat.js

@@ -1,6 +1,7 @@
 import { DataFrame } from './frame'
 // import { Series } from "../../src/core/series";
 import { Utils } from "./utils"
+import { Series } from './series'
 
 const utils = new Utils()
 
@@ -88,9 +89,19 @@ export class Concat {
                     let val = values[index]
                     if (typeof data[index] === "undefined") {
 
-                        data[index] = val;
+                        if(Array.isArray(val)){
+                            data[index] = val
+                        }else{
+                            data[index] = [val];
+                        }
+                        
                     } else {
-                        data[index].push(...val);
+                        if(Array.isArray(val)){
+                            data[index].push(...val);
+                        }else{
+                            data[index].push(val);
+                        }
+                        
                     }
                 }
 
@@ -146,7 +157,6 @@ export class Concat {
                         not_exist.push(col_name);
                     }
                 }
-
                 if (not_exist.length > 0) {
                     for (let i = 0; i < value.length; i++) {
                         let row_value = value[i]
@@ -160,7 +170,12 @@ export class Concat {
                                 new_arr[j] = NaN
                             } else {
                                 let index = df_columns.indexOf(col_name)
-                                new_arr[j] = row_value[index]
+                                if(Array.isArray(row_value)){
+                                    new_arr[j] = row_value[index]
+                                }else{
+                                    new_arr[j] = row_value;
+                                }
+                                
                             }
 
                         }
@@ -171,9 +186,16 @@ export class Concat {
                 }
 
             }
+            
+            if(Array.isArray(data[0])){
+                let df = new DataFrame(data, { columns: columns });
+                return df;
+            }else{
+                let sf = new Series(data)
+                return sf
+            }
 
-            let df = new DataFrame(data, { columns: columns });
-            return df;
+            
 
         }
     }

danfojs/tests/core/concat.js

@@ -1,6 +1,7 @@
 import { assert } from "chai"
 import { DataFrame } from '../../src/core/frame'
 import {concat} from '../../src/core/concat'
+import { Series } from "../../src/core/series";
 
 
 
@@ -48,4 +49,89 @@ describe("Concatenate", function () {
             NaN, NaN, NaN]]
         assert.deepEqual(new_df.values, data_values);
     });
+
+    it("concatenate dataframe and series along 0 axis",function(){
+
+        let data1 = [1,2,3,4]
+        let data2 = [3,4,5,6]
+
+        let data = [[1, 2, 3], [4, 5, 6], [20, 30, 40], [39, 89, 78]]
+        let cols = ["A", "B", "C"]
+        let df = new DataFrame(data, { columns: cols })
+
+        let s1 = new Series(data1)
+        let s2 = new Series(data2)
+        let rslt = [
+            [ 1, 2, 3, NaN ],
+            [ 4, 5, 6, NaN ],
+            [ 20, 30, 40, NaN ],
+            [ 39, 89, 78, NaN ],
+            [ NaN, NaN, NaN, 1 ],
+            [ NaN, NaN, NaN, 2 ],
+            [ NaN, NaN, NaN, 3 ],
+            [ NaN, NaN, NaN, 4 ]
+          ]
+           
+
+        let con = concat({ "df_list": [df,s1], "axis": 0 })
+
+        assert.deepEqual(con.values, rslt)
+
+    })
+
+    it("concatenate dataframe and series along axis 1",function(){
+
+        let data = [[1, 2, 3], [4, 5, 6], [20, 30, 40], [39, 89, 78]]
+        let cols = ["A", "B", "C"]
+        let df = new DataFrame(data, { columns: cols })
+
+        let data1 = [1,2,3,4]
+        let s1 = new Series(data1)
+        let rslt = [
+            [ 1, 2, 3, 1 ],
+            [ 4, 5, 6, 2 ],
+            [ 20, 30, 40, 3 ],
+            [ 39, 89, 78, 4 ]
+          ]
+           
+
+        let con = concat({ "df_list": [df,s1], "axis": 1 })
+
+        assert.deepEqual(con.values, rslt)
+
+    })
+    it("concatenate series along axis 1",function(){
+
+        let data1 = [1,2,3,4]
+        let data2 = [3,4,5,6]
+
+        let s1 = new Series(data1)
+        let s2 = new Series(data2)
+        let rslt = [ [ 1, 3 ], [ 2, 4 ], [ 3, 5 ], [ 4, 6 ] ]
+           
+
+        let con = concat({ "df_list": [s1,s2], "axis": 1 })
+
+        assert.deepEqual(con.values, rslt)
+
+    })
+    it("concatenate series along axis 0",function(){
+
+        let data1 = [1,2,3,4]
+        let data2 = [3,4,5,6]
+
+        let s1 = new Series(data1)
+        let s2 = new Series(data2)
+        let rslt = [
+            1, 2, 3, 4,
+            3, 4, 5, 6
+          ]
+           
+        let con = concat({ "df_list": [s1,s2], "axis": 0 })
+
+        assert.deepEqual(con.values, rslt)
+
+    })
+
+
 });