How can I group a stacked bar chart?

by

The following approach allows grouped and stacked bars at the same time.
First the dataframe is sorted by parameter, context. Then the context is unstacked from the index, creating new columns for every context, value pair.
Finally, three bar plots are drawn over each other to visualize the stacked bars.

import pandas as pd
from matplotlib import pyplot as plt

df = pd.DataFrame(columns=['Context', 'Parameter', 'Val1', 'Val2', 'Val3'],
                  data=[['Total', 1, 43.312347, 9.507902, 1.580367],
                        ['Total', 2, 42.862649, 9.482205, 1.310549],
                        ['Total', 3, 43.710651, 9.430811, 1.400488],
                        ['Total', 4, 43.209559, 9.803418, 1.349094],
                        ['Total', 5, 42.541436, 10.008994, 1.220609],
                        ['Total', 6, 42.978286, 9.430811, 1.336246],
                        ['Total', 7, 42.734164, 10.317358, 1.606064],
                        ['User', 1, 47.652348, 11.138861, 2.297702],
                        ['User', 2, 47.102897, 10.589411, 1.848152],
                        ['User', 3, 46.853147, 10.139860, 1.848152],
                        ['User', 4, 47.252747, 11.138861, 1.748252],
                        ['User', 5, 45.954046, 10.239760, 1.448551],
                        ['User', 6, 46.353646, 10.439560, 1.498501],
                        ['User', 7, 47.102897, 11.338661, 1.998002]])
df.set_index(['Context', 'Parameter'], inplace=True)
df0 = df.reorder_levels(['Parameter', 'Context']).sort_index()

colors = plt.cm.Paired.colors

df0 = df0.unstack(level=-1) # unstack the 'Context' column
fig, ax = plt.subplots()
(df0['Val1']+df0['Val2']+df0['Val3']).plot(kind='bar', color=[colors[1], colors[0]], rot=0, ax=ax)
(df0['Val2']+df0['Val3']).plot(kind='bar', color=[colors[3], colors[2]], rot=0, ax=ax)
df0['Val3'].plot(kind='bar', color=[colors[5], colors[4]], rot=0, ax=ax)

legend_labels = [f'{val} ({context})' for val, context in df0.columns]
ax.legend(legend_labels)

plt.tight_layout()
plt.show()

resulting bar plot

More Related Contents:

  1. How to plot in multiple subplots
  2. How to have clusters of stacked bars with python (Pandas)
  3. Python matplotlib multiple bars
  4. plot different color for different categorical levels using matplotlib
  5. How to plot multiple Seaborn Jointplot in Subplot
  6. Color by Column Values in Matplotlib
  7. Plotting grouped data in same plot using Pandas
  8. How to add value labels on a bar chart
  9. How to add hovering annotations to a plot
  10. How to draw vertical lines on a given plot
  11. Plot a horizontal line on a given plot
  12. How to have clusters of stacked bars
  13. How to plot multiple dataframes in subplots
  14. How to plot and annotate grouped bars in seaborn / matplotlib
  15. Seaborn showing scientific notation in heatmap for 3-digit numbers
  16. Seaborn: countplot() with frequencies
  17. multi index plotting
  18. How do you plot a vertical line on a time series plot in Pandas?
  19. How to remove or hide x-axis labels from a seaborn / matplotlib plot
  20. Plotting time-series data with seaborn
  21. How to deal with NaN values when plotting a boxplot
  22. How to edit properties of whiskers, fliers, caps, etc. in Seaborn boxplot
  23. Using pandas crosstab to create a bar plot
  24. How to annotate a seaborn barplot with the aggregated value
  25. Discrete legend in seaborn heatmap plot
  26. plot different color for different categorical levels
  27. countplot() with frequencies
  28. Add image annotations to bar plots
  29. Seaborn multiple barplots
  30. Add density curve on the histogram

Leave a Comment