Given a pandas df I want to create Stacked Bar Graphs where all the values per row are stacked in each bar. I want the xticks to be the index number and the y value to be the sum of the stacked bars of each row. However I haven´t been able to achieve it.
I get TypeError: only size-1 arrays can be converted to Python scalars when I try to do the plot
I've tried to append each row in an array but I end up appending the same arrange multiple times.
I'm following the example here: https://matplotlib.org/3.1.1/gallery/lines_bars_and_markers/bar_stacked.html#stacked-bar-graph
import pandas as pd
import matplotlib as plt
index C1 C2 C3
1 48692.4331 34525.0003 14020.1233
2 43206.1635 27978.9984 16572.0428
3 67398.4482 49903.4956 29856.5693
no_1 = [df["C1"] for index in df.index]
no_2 = [df["C2"] for index in df.index]
no_3 = [df["C3"] for index in df.index]
N = 3
ind = np.arange(N) # the x locations for the groups
width = 0.35 # the width of the bars: can also be len(x) sequence
p1 = plt.bar(ind, no_1, width)
p2 = plt.bar(ind, no_2, width, bottom=no_1)
p3 = plt.bar(ind, no_3, width, bottom=no_2)
plt.xticks(ind, ('no_1', 'no_2', 'no_3'))
You could use pandas.DataFrame.plot
:
df.rename(lambda x: 'no_'+str(x), axis='index').plot.bar(stacked=True)
Output:
For learning purposes:
xlabels = 'no_'+ df.index.astype(str)
_ = plt.bar(xlabels, df['C1'])
_ = plt.bar(xlabels, df['C2'], bottom=df['C1'])
_ = plt.bar(xlabels, df['C3'], bottom=df[['C1','C2']].sum(1))
Output:
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.