From 9a5d812ee608fc07d3c0303deee88cd20bc92125 Mon Sep 17 00:00:00 2001 From: Ville Brofeldt <33317356+villebro@users.noreply.github.com> Date: Thu, 23 Jul 2020 09:22:48 +0300 Subject: [PATCH] fix: pie chart multiple groupbys (#10391) --- superset/viz.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/superset/viz.py b/superset/viz.py index 8bceef675..6ce4d5ac9 100644 --- a/superset/viz.py +++ b/superset/viz.py @@ -1527,10 +1527,10 @@ class DistributionPieViz(NVD3Viz): 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")