I currently have a dataframe (df) which looks like this: and I need to graph it so it looks like this:
![Desired graph][2]
Where in the graph, the columns are 'index', the blue bars are 'Total' and orange bars are 'Total.1'
And every time I've tried to do it I get KeyErrors, can someone PLEASE point me in the right direction because I feel like I've tried everything but something always goes wrong
To generate a simple horizontal stacked bar chart is easy, the below code will accomplish the task.
df.set_index('index').plot(stacked=True, kind='barh')
plt.tight_layout()
plt.show()
However, since your labels are so long, the layout will cut off information, and you may want to pre-process your labels so they look better.
Here is what I would recommend in terms of plot format:
options = {
'stacked': True,
'kind': 'barh',
'title': 'Medals at the Winter and Summer Olympics'
}
df.rename(columns={'Total': 'Winter Games', 'Total.1': 'Summer Games'}).assign(
code=df['index'].str.extract(r'\((.*?)\)')
).set_index('code')[['Winter Games', 'Summer Games']].plot(**options)
plt.show()
Which results in:
