create ndFrame from tensor

Author: risenW

danfojs/src/core/generic.js

@@ -25,7 +25,9 @@ export default class NDframe {
 
     constructor(data, kwargs = {}) {
         this.kwargs = kwargs
-
+        if (data instanceof tf.Tensor){
+            data = data.arraySync()
+        }
         if (utils.__is_1D_array(data)) {
             this.series = true
             this.__read_array(data)

danfojs/tests/core/generic.js

@@ -1,5 +1,7 @@
 import { assert } from "chai";
 import NDframe from '../../src/core/generic';
+import * as tf from '@tensorflow/tfjs'
+
 
 
 describe("Generic (NDFrame)", function () {
@@ -97,6 +99,23 @@ describe("Generic (NDFrame)", function () {
         })
     })
 
+    describe("NDframe Created from a Tensor", function () {
+
+        it("prints the shape of a 2D frame created from a 2D tensor", function () {
+            let data = tf.tensor([1, 2, 3, 4])
+            let ndframe = new NDframe(data)
+            assert.deepEqual(ndframe.ndim, 1)
+            assert.deepEqual(ndframe.values, [1, 2, 3, 4])
+
+        })
+        it("prints the shape of a 2D frame created from a 1D tensor", function () {
+            let data = tf.tensor([[2, 3, 4], [4, 5, 6]])
+            let ndframe = new NDframe(data, {columns: ["alpha", "count", "sum"]})
+            assert.deepEqual(ndframe.column_names, ["alpha", "count", "sum"])
+        })
+       
+    })
+
     describe("index", function () {
         it("Returns the index of an NDframe", function () {
             let data = [{ alpha: "A", count: 1 }, { alpha: "B", count: 2 }, { alpha: "C", count: 3 }]
@@ -116,12 +135,12 @@ describe("Generic (NDFrame)", function () {
             let data = { alpha: ["A", "B", "C", "D"], count: [1, 2, 3, 4] }
             let ndframe = new NDframe(data)
             // assert.deepEqual(ndframe.shape, [4, 2])
-            assert.deepEqual(ndframe.col_data, [["A", "B", "C", "D"],[1, 2, 3, 4]])
+            assert.deepEqual(ndframe.col_data, [["A", "B", "C", "D"], [1, 2, 3, 4]])
         })
         it("retrieves the col data ", function () {
-            let data = { alpha: ["A", "B", "C", "D"], count: [1,2,3,4], sum: [20.3, 30.456, 40.90, 90.1]}
+            let data = { alpha: ["A", "B", "C", "D"], count: [1, 2, 3, 4], sum: [20.3, 30.456, 40.90, 90.1] }
             let ndframe = new NDframe(data)
-            let res = [["A", "B", "C", "D"],[1,2,3,4],[20.3, 30.456, 40.90, 90.1]]
+            let res = [["A", "B", "C", "D"], [1, 2, 3, 4], [20.3, 30.456, 40.90, 90.1]]
             assert.deepEqual(ndframe.col_data, res)
         })
         it("retrieves the row data created from OA ", function () {
@@ -183,7 +202,7 @@ describe("Generic (NDFrame)", function () {
             assert.deepEqual(ndframe.dtypes, ["float32"])
         })
 
-       
+
         it("Returns dtype set during creation of 2DFrame from an Object", function () {
             let data = [{ alpha: "A", count: 1 }, { alpha: "B", count: 2 }]
             let options = { dtypes: ['string', 'int32'] }
@@ -214,13 +233,13 @@ describe("Generic (NDFrame)", function () {
             let data = [{ alpha: "A", count: 1 }, { alpha: "B", count: 2 }, { alpha: "C", count: 3 }]
             let df = new NDframe(data)
             let result = `alpha,count\nA,1\nB,2\nC,3\n`
-            df.to_csv().then((csv)=>{
+            df.to_csv().then((csv) => {
                 assert.deepEqual(csv, result)
             })
         })
         it("Converts DataFrame of Series to csv format and return string when path is not specified", async function () {
             let data = [[12, 2, 20], [90, 5, 23], [45, 56, 70], [9, 10, 19]]
-            let df = new NDframe(data, {columns: ["A", "B",  "C"]})
+            let df = new NDframe(data, { columns: ["A", "B", "C"] })
             let result = `A,B,C\n12,2,20\n90,5,23\n45,56,70\n9,10,19\n`
             assert.deepEqual(await df.to_csv(), result)
         })
@@ -232,15 +251,15 @@ describe("Generic (NDFrame)", function () {
             let result = JSON.stringify([{ alpha: "A", count: 1 }, { alpha: "B", count: 2 }, { alpha: "C", count: 3 }])
 
             let df = new NDframe(data)
-            df.to_json().then((json)=>{
+            df.to_json().then((json) => {
                 assert.deepEqual(json, result)
             })
         })
         it("Converts DataFrame to json format", async function () {
             let data = [[12, 2, 20], [90, 5, 23], [45, 56, 70]]
-            let df = new NDframe(data, {columns: ["A", "B",  "C"]})
-            let result = JSON.stringify([{A: 12, B: 2, C: 20},{A: 90, B: 5, C: 23},{A: 45, B: 56, C: 70}])
-            df.to_json().then((json)=>{
+            let df = new NDframe(data, { columns: ["A", "B", "C"] })
+            let result = JSON.stringify([{ A: 12, B: 2, C: 20 }, { A: 90, B: 5, C: 23 }, { A: 45, B: 56, C: 70 }])
+            df.to_json().then((json) => {
                 assert.deepEqual(json, result)
             })
         })

package.json

@@ -23,7 +23,7 @@
     "table": "^5.4.6"
   },
   "scripts": {
-    "test": "nyc mocha --require @babel/register danfojs/tests/core/utils",
+    "test": "nyc mocha --require @babel/register danfojs/tests/core/generic",
     "dev": "npm run lint && babel ./danfojs/src -d dist --no-comments",
     "build": "babel ./danfojs/src -d ./dist --no-comments",
     "lint": "eslint ./danfojs/src",