add combine config for metrics in pivot table (#3086)

* add combine config for metrics in pivot table * change method to stack/unstack * update backendSync
apache · Jul 28, 2017 · b58cfbc · b58cfbc
1 parent 1e325d9
commit b58cfbc
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 1 deletion.
diff --git a/superset/assets/backendSync.json b/superset/assets/backendSync.json
@@ -750,6 +750,12 @@
       "default": false,
       "description": "Sort bars by x labels."
     },
+    "combine_metric": {
+      "type": "CheckboxControl",
+      "label": "Combine Metrics",
+      "default": false,
+      "description": "Display metrics side by side within each column, as opposed to each column being displayed side by side for each metric."
+    },
     "show_controls": {
       "type": "CheckboxControl",
       "label": "Extra Controls",

diff --git a/superset/assets/javascripts/explore/stores/controls.jsx b/superset/assets/javascripts/explore/stores/controls.jsx
@@ -249,6 +249,14 @@ export const controls = {
     description: 'Sort bars by x labels.',
   },
 
+  combine_metric: {
+    type: 'CheckboxControl',
+    label: 'Combine Metrics',
+    default: false,
+    description: 'Display metrics side by side within each column, as ' +
+    'opposed to each column being displayed side by side for each metric.',
+  },
+
   show_controls: {
     type: 'CheckboxControl',
     label: 'Extra Controls',

diff --git a/superset/assets/javascripts/explore/stores/visTypes.js b/superset/assets/javascripts/explore/stores/visTypes.js
@@ -337,7 +337,7 @@ export const visTypes = {
         controlSetRows: [
           ['groupby', 'columns'],
           ['metrics', 'pandas_aggfunc'],
-          ['number_format', 'pivot_margins'],
+          ['number_format', 'combine_metric', 'pivot_margins'],
         ],
       },
     ],

diff --git a/superset/viz.py b/superset/viz.py
@@ -402,6 +402,9 @@ def get_data(self, df):
             aggfunc=self.form_data.get('pandas_aggfunc'),
             margins=self.form_data.get('pivot_margins'),
         )
+        # Display metrics side by side with each column
+        if self.form_data.get('combine_metric'):
+            df = df.stack(0).unstack()
         return dict(
             columns=list(df.columns),
             html=df.to_html(