Pandas/Matplotlib - How to sort values in a plot

Question

I have a dataset with columns "Company" and "Outcome".

Company is a company name, Outcome is either success or failure

I have created the following graph with this code. How can I sort this so that the top has the most "Outcomes" (i.e. combined Success and Failure) and it goes down in a descending manner?

The code I used is:

df = pd.DataFrame({'Company': ['C', 'A', 'A', 'B', 'B', 'B', 'B'], 'Outcome': ['Success', 'Success', 'Failure', 'Failure', 'Success', 'Failure','Success' ]})

df.groupby(['Company', 'Outcome']).size().unstack().plot(kind = 'barh', stacked=True)

plt.show()

Additionally, including sort_values() after size() does not appear to have any effect, so clearly I am using it wrong. Any advice?

Bar Graph

Please add a dummy dataframe. Also, when you use the sort_values, what are you sorting on? 'Company'? If so it would do it alphabetically. — le_camerone
– le_camerone, Commented Apr 4, 2022 at 2:35
I'm new here - I've just updated the code to a simple example. The dummy dataframe isn't showing up despite being in the code - df = pd.DataFrame({'Company': ['C', 'A', 'A', 'B', 'B', 'B', 'B'], 'Outcome': ['Success', 'Success', 'Failure', 'Failure', 'Success', 'Failure','Success' ]}) — Haris
– Haris, Commented Apr 4, 2022 at 2:57

r-beginners · Accepted Answer · 2022-04-04 05:16:22Z

1

As noted in the comments, sorting at the point of data will sort from the bottom to the largest, so the y-axis should be in reverse order.

Added: Need to sort by total value

df2 = pd.DataFrame({'Company': ['C', 'A', 'A', 'B', 'B', 'B', 'B', 'D', 'D', 'D', 'D', 'D', 'D'],
                    'Outcome': ['Success', 'Success', 'Failure', 'Failure', 'Success', 'Failure','Success','Success','Success','Success','Success','Success','Success' ]})

df3 = df2.groupby(['Company', 'Outcome']).size().unstack()
df3['Total'] = df3['Failure'].fillna(0)+ df3['Success']
df3.sort_values('Total', ascending=False, inplace=True)
ax = df3[['Failure','Success']].plot(kind='barh', stacked=True)
ax.invert_yaxis()

edited Apr 4, 2022 at 5:16

answered Apr 4, 2022 at 4:27

r-beginners

35.6k3 gold badges20 silver badges41 bronze badges

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

3 Comments

Haris Over a year ago

This is part of the way there. I've applied to a slightly bigger dataset and it appears to be sorting by the number of Successes first, and then subsequently by the number of Failures. I was hoping to get it to sort by total number of successes + failures. For example, it doesn't show this neatly arranged in terms of total - df = pd.DataFrame({'Company': ['C', 'A', 'A', 'B', 'B', 'B', 'B', 'D', 'D', 'D', 'D', 'D', 'D'], 'Outcome': ['Success', 'Success', 'Failure', 'Failure', 'Success', 'Failure','Success','Success','Success','Success','Success','Success','Success' ]})

r-beginners Over a year ago

If you need to sort by total, add a new total column and then sort by total column.

Haris Over a year ago

How do I assign the appropriate values to the Total column?

Collectives™ on Stack Overflow

Pandas/Matplotlib - How to sort values in a plot

1 Answer 1

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related