Marimekko aka Mekko plot #4019

Activity

1. nicolaskruchten

   added

   type: new trace type

   on Jul 5, 2019
2. eddy-geek commented on Apr 9, 2020

   

   eddy-geek

   on Apr 9, 2020

   Last edited by eddy-geek

   More actions

   Here is an implementation based on @nicolaskruchten work on #4461 :

   def stacked_bar_width_plot(df, value_cols, width_col, colors=None, **subplot):
       """A stacked column plot with variable bar width.
          :param df
          :param list value_cols: columns of `df`, already normalized (sum=1 for every line).
          :param str width_col: column of `df`, unbounded, used (i) as label (ii) to compute width.
          :param dict subplot: optional figure/row/col
       """
       categories = df.index.to_list()
       width = df[width_col]
       htmpl = width_col + ': %{customdata}, %{y}'
       x = np.cumsum([0] + list(width[:-1]))
       colors = colors or [None, ] * len(value_cols)
   
       figure = subplot.pop('figure', go.Figure())
       for colname, color in zip(value_cols, colors):
           figure.add_trace(go.Bar(name=colname, x=x, y=df[colname], width=width, hovertemplate=htmpl, customdata=width, offset=0, marker_color=color), **subplot)
       return figure.update_xaxes(
           tickvals=x + np.array(width) / 2,
           ticktext=categories,
           range=[0, np.sum(width)], **subplot
       ) \
           .update_yaxes(tickformat=',.0%', range=[0, 1], row=subplot.get('row'), col=subplot.get('col')) \
           .update_layout(barmode='stack', bargap=0)
   
   
   def mekko_plot(df, unit_name=None, colors=None, **subplot):
       """A mekko plot is a normalized stacked column plot with variable bar width.
   
          :param DataFrame df: already indexed by category, with only numeric columns:
                      there will be one stacked-bar per line (X), labeled after the index, with one bar per column.
          :param str unit_name: used to populate hover.
          :param list colors: color of each column. None for default palette (blue, red, ...).
          The rest (title..) is easily added afterwards.
       """
       # Normalize then defer to stacked_bar_width_plot plot.
       value_cols = df.columns
       w = pd.DataFrame({unit_name: df.sum(axis='columns')})
       w[value_cols] = df.div(w[unit_name], axis='index')
   
       return stacked_bar_width_plot(w, value_cols, width_col=unit_name, colors=colors, **subplot)
   
   # Example
   df = pd.DataFrame(dict(category=['a','a', 'b',  'b', 'a','a', 'b',  'b'],
                          var1=[9,2,5,3,2,4,6,0],
                          var2=[1,2,3,4,2,6,7,8])).set_index('category')
   display(df)
   mekko_plot(df, unit_name='some_label', colors=['blue', 'red'])

   Display:
   

   👍React with 👍3Reacted by J S, Víctor Poblete Ramírez and Bruce-at-Artefact
3. prodigeek commented on Dec 31, 2020

   

   prodigeek

   on Dec 31, 2020

   More actions

   Is there a way to assign a single color to a column in this code? I have a large mekko with many individual stacks - I'd rather have each column have a single color. Any suggestions?
   Thanks!
4. vpobleteacustica commented on Dec 4, 2021

   

   vpobleteacustica

   on Dec 4, 2021

   More actions

   Many thanks, great job!
5. gvwilson commented on Jun 26, 2024

   

   gvwilson

   on Jun 26, 2024

   Contributor

   More actions

   Hi - we are trying to tidy up the stale issues and PRs in Plotly's public repositories so that we can focus on things that are still important to our community. Since this one has been sitting for several years, I'm going to close it; if it is still a concern, please add a comment letting us know what recent version of our software you've checked it with so that I can reopen it and add it to our backlog. Thanks for your help - @gvwilson
6. gvwilson

   closed this as completedon Jun 26, 2024