Add countplot to Seaborn

minjk-bl · minjk-bl · commit 430f16ceb479 · 2022-05-19T12:20:50.000+09:00
diff --git a/data/m_visualize/seabornLibrary.js b/data/m_visualize/seabornLibrary.js
@@ -19,7 +19,7 @@ define([
      *      }
      * ]
      */
-    var CHART_LIBRARIES = {
+    var SEABORN_LIBRARIES = {
         /** Relational plots */
         'scatterplot': {
             name: 'Scatter Plot',
@@ -155,7 +155,19 @@ define([
                 { name: 'allocateTo', label: 'Allocate To', component: ['input'], usePair: true }
             ]
         },
+        'countplot': {
+            name: 'Count Plot',
+            code: '${allocateTo} = sns.countplot(${data}${x}${y}${hue}${etc})',
+            description: 'Show the counts of observations in each categorical bin using bars.',
+            options: [
+                { name: 'data', component: ['var_select'], var_type: ['DataFrame', 'Series', 'list'], usePair: true },
+                { name: 'x', component: ['col_select'], usePair: true },
+                { name: 'y', component: ['col_select'], usePair: true },
+                { name: 'hue', component: ['col_select'], usePair: true },
+                { name: 'allocateTo', label: 'Allocate To', component: ['input'], usePair: true }
+            ]
+        }
     }
 
-    return CHART_LIBRARIES;
+    return SEABORN_LIBRARIES;
 });
diff --git a/js/m_visualize/Seaborn.js b/js/m_visualize/Seaborn.js
@@ -21,8 +21,8 @@ define([
     'vp_base/js/com/component/PopupComponent',
     'vp_base/js/com/component/SuggestInput',
     'vp_base/js/com/component/VarSelector2',
-    'vp_base/data/m_visualize/chartLibrary'
-], function(chartHTml, chartCss, com_String, com_generator, com_util, PopupComponent, SuggestInput, VarSelector2, CHART_LIBRARIES) {
+    'vp_base/data/m_visualize/seabornLibrary'
+], function(chartHTml, chartCss, com_String, com_generator, com_util, PopupComponent, SuggestInput, VarSelector2, SEABORN_LIBRARIES) {
 
     class Seaborn extends PopupComponent {
         _init() {
@@ -67,12 +67,12 @@ define([
                 ...this.state
             }
             
-            this.chartConfig = CHART_LIBRARIES;
+            this.chartConfig = SEABORN_LIBRARIES;
             this.chartTypeList = {
                 'Relational': [ 'scatterplot', 'lineplot' ],
                 'Distributions': [ 'histplot', 'kdeplot', 'rugplot' ], 
                 'Categorical': [ 'stripplot', 'swarmplot', 'boxplot', 'violinplot', 'pointplot', 'barplot' ],
-                // 'ETC': [ ]
+                'ETC': [ 'countplot' ]
             }
 
             this.legendPosList = [
