Plotly python bar plot stack order

Question

Here is my code for the dataframe

df = pd.DataFrame({'var_1':['a', 'a', 'a', 'b', 'b', 'b', 'c', 'c', 'c'], 
                   'var_2':['m', 'n', 'o', 'm', 'n', 'o', 'm', 'n', 'o'], 
                   'var_3':[np.random.randint(25, 33) for _ in range(9)]})

Here is the dataframe that I have

var_1   var_2   var_3
0   a   m   27
1   a   n   28
2   a   o   28
3   b   m   31
4   b   n   30
5   b   o   25
6   c   m   27
7   c   n   32
8   c   o   27

Here is the code I used to get the stacked bar plot

fig = px.bar(df, x='var_3', y='var_1', color='var_2', orientation='h', text='var_3')
fig.update_traces(textposition='inside', insidetextanchor='middle')
fig

But I want the bar to stack in descending order of the values, largest at the start/bottom and smallest at top
How should I update the layout to get that

To clarify, you want largest var_3 at bottom? (and you intend for the colours to demark separate entities too?) So bottom would be largest of m + n + 0 for any particular var_1? — Amiga500
– Amiga500, Commented Apr 19, 2021 at 16:27
I don't want m-n-o in order. Which ever is the largest, it should be at the begining If m is the largest, then it should be at the start, then the second-largest, and then the third in the stacking order — imdevskp
– imdevskp, Commented Apr 19, 2021 at 16:34

Kharlanov Sergey · Accepted Answer · 2021-04-19 16:46:03Z

1

df = pd.DataFrame({'var_1':['a', 'a', 'a', 'b', 'b', 'b', 'c', 'c', 'c'], 
                   'var_2':['m', 'n', 'o', 'm', 'n', 'o', 'm', 'n', 'o'], 
                   'var_3':[np.random.randint(25, 33) for _ in range(9)]})

df.sort_values(['var_1', 'var_3'], ignore_index=True, inplace=True, ascending=False)

# colors
colors = {'o': 'red',
          'm': 'blue',
          'n': 'green'}

# traces
data = []

# loop across the different rows
for i in range(df.shape[0]):
        data.append(go.Bar(x=[df['var_3'][i]],
                           y=[df['var_1'][i]],
                           orientation='h',
                           text=str(df['var_3'][i]),
                           marker=dict(color=colors[df['var_2'][i]]),
                           name=df['var_2'][i],
                           legendgroup=df['var_2'][i],
                           showlegend=(i in [1, 2, 3])))
        
# layout
layout = dict(barmode='stack', 
              yaxis={'title': 'var_1'},
              xaxis={'title': 'var_3'})

# figure
fig = go.Figure(data=data, layout=layout)    
fig.update_traces(textposition='inside', insidetextanchor='middle')
fig.show()

edited Apr 19, 2021 at 16:46

answered Apr 19, 2021 at 16:04

Kharlanov Sergey

363 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Amiga500 Over a year ago

Yep - I was thinking the data would have to be flattened, haven't found a built in workaround yet.

Collectives™ on Stack Overflow

Plotly python bar plot stack order

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related