fix: pie chart multiple groupbys (#10391)

apache · Sep 5, 2020 · 1c7022e · 1c7022e
1 parent 003161e
commit 1c7022e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/superset/viz.py b/superset/viz.py
@@ -1547,10 +1547,10 @@ def get_data(self, df: pd.DataFrame) -> VizData:
         if df.empty:
             return None
         metric = self.metric_labels[0]
-        df = df.pivot_table(index=self.groupby, values=[metric])
-        df.sort_values(by=metric, ascending=False, inplace=True)
-        df = df.reset_index()
-        df.columns = ["x", "y"]
+        df = pd.DataFrame(
+            {"x": df[self.groupby].agg(func=", ".join, axis=1), "y": df[metric]}
+        )
+        df.sort_values(by="y", ascending=False, inplace=True)
         return df.to_dict(orient="records")