update plot type methods

Author: risenW

danfojs/src/plotting/plot.js

@@ -1,6 +1,8 @@
 //perform plotting
 import { newPlot } from 'plotly.js/dist/plotly'
 import { Utils } from "../core/utils"
+import { Series } from "../core/series"
+
 const utils = new Utils()
 
 
@@ -15,31 +17,42 @@ export class Plot {
         let params = Object.keys(config)
         let this_config = {}
 
-        if (!utils.__key_in_object(config, "layout")) {
-            this_config['layout'] = {}
-        }
-
         params.forEach(param => {
             this_config[param] = config[param]
         })
 
-        if (ndframe.series) {
-            let x = ndframe.values
+        if (!utils.__key_in_object(config, "layout")) {
+            this_config['layout'] = {}
+        }
+
+        if (ndframe instanceof Series) {
             let trace = {}
-            trace["x"] = x
+
+            if (this_config['type'] == 'histogram') {
+                let x = ndframe.values
+                trace["x"] = x
+            } else {
+                let y = ndframe.values
+                trace["y"] = y
+            }
 
             params.forEach(param => {
-                if (!param == 'layout') {
-                    trace[param] = config[param]
-                }
+                trace[param] = config[param]
             })
-
-
             newPlot(div, [trace], this_config['layout']);
 
         } else {
+
             //DataFrame
             if (utils.__key_in_object(this_config, 'x') && utils.__key_in_object(this_config, 'y')) {
+                if (!ndframe.column_names.includes(this_config['x'])) {
+                    throw Error(`Column Error: ${this_config['x']} not found in columns`)
+                }
+                if (!ndframe.column_names.includes(this_config['y'])) {
+                    throw Error(`Column Error: ${this_config['y']} not found in columns`)
+                }
+
+
                 let x = ndframe[this_config['x']].values
                 let y = ndframe[this_config['y']].values
 
@@ -60,18 +73,86 @@ export class Plot {
 
                 newPlot(div, [trace], this_config['layout']);
 
+
+            } else if (this_config['type'] == 'pie') {
+                if (!ndframe.column_names.includes(this_config['values'])) {
+                    throw Error(`Column Error: ${this_config['values']} not found in columns`)
+                }
+                if (!ndframe.column_names.includes(this_config['labels'])) {
+                    throw Error(`Column Error: ${this_config['labels']} not found in columns`)
+                }
+                let data = [{
+                    values: ndframe[this_config['values']].values,
+                    labels: ndframe[this_config['labels']].values,
+                    type: 'pie'
+                }];
+
+                newPlot(div, data, this_config['layout'])
+
+            } else if (this_config['type'] == 'table') {
+                let header = {}
+                let cells = {}
+
+                header['values'] = ndframe.column_names
+                cells['values'] = ndframe.col_data
+
+                if (this_config['header_style']) {
+                    Object.keys(this_config['header_style']).forEach(param => {
+                        header[param] = this_config['header_style'][param]
+                    })
+                }
+
+                if (this_config['cell_style']) {
+                    Object.keys(this_config['cell_style']).forEach(param => {
+                        header[param] = this_config['cell_style'][param]
+                    })
+                }
+                var data = [{
+                    type: 'table',
+                    header: header,
+                    cells: cells
+                }]
+                newPlot(div, data, this_config['layout']);
+
+            } else if (this_config['type'] == 'box') {
+                let cols_2_show = []
+                let data = []
+
+                if (this_config['columns'] != undefined) {
+                    cols_2_show = this_config['columns']
+                    cols_2_show.forEach(col => {
+                        if (!ndframe.column_names.includes(col)) {
+                            throw Error(`Column Error: ${this_config['labels']} not found in columns`)
+                        }
+                    })
+                } else {
+                    cols_2_show = ndframe.column_names
+                }
+
+
+                cols_2_show.forEach(col => {
+                    let col_idx = ndframe.column_names.indexOf(col)
+                    let trace = []
+                    trace['y'] = ndframe.col_data[col_idx]
+                    trace["type"] = 'box'
+                    trace["name"] = col
+
+                    data.push(trace)
+                })
+                newPlot(div, data, this_config['layout']);
+
             } else {
                 //plot all
-                let y = ndframe.index
+                let x = ndframe.index
                 let data = []
                 ndframe.column_names.forEach(c_name => {
-                    let trace = {
-                        x: ndframe[c_name].values,
-                        y: y,
-                        type: this_config["type"],
-                        mode: this_config["mode"],
-                        name: c_name
-                    };
+                    let trace = {}
+                    trace["x"] = x
+                    trace["y"] = ndframe[c_name].values
+                    trace['name'] = c_name
+                    params.forEach(param => {
+                        trace[param] = config[param]
+                    })
                     data.push(trace)
 
                 })