Matplotlib plotting dataframe

Question

I'm trying to learn how to plot dataframes. I read in a csv and have the following columns:

 cost, model, origin, year
--------------------------
200    x1     usa     2020
145    x1     chn     2020
233    x1     usa     2020
122    x2     chn     2020
583    x2     usa     2020
233    x3     chn     2020 
201    x3     chn     2020

I'm trying to create a bar plot and only want to plot the average cost per model.

Here's my attempt, but I dont think im on the right track:

df = df.groupby('cost').mean()
plt.bar(df.index, df['model'])
plt.show()

Can you provide an example of the data frame?

Rithwik Babu
– Rithwik Babu

2021-04-05 17:51:24 +00:00
Commented Apr 5, 2021 at 17:51 — Rithwik Babu
– Rithwik Babu, Commented Apr 5, 2021 at 17:51
@RithwikBabu let me know if that helps

Leslie Tate
– Leslie Tate

2021-04-05 17:58:08 +00:00
Commented Apr 5, 2021 at 17:58 — Leslie Tate
– Leslie Tate, Commented Apr 5, 2021 at 17:58

perl · Accepted Answer · 2021-04-05 18:06:35Z

1

You can groupby model, then calculate the mean of cost and plot it:

df.groupby('model')['cost'].mean().plot.bar()

Output:

Or with seaborn:

sns.barplot(data=df, x='model', y='cost', ci=None)

Output:

answered Apr 5, 2021 at 18:06

perl

9,9811 gold badge14 silver badges23 bronze badges

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

5 Comments

Leslie Tate Over a year ago

thank you for providing this. So question ontop of this. Lets say theres 100+ models. If i wanted to take the head(20) and sort in descending order. How would i do that?

perl Over a year ago

Sort in descending order by average cost or by the name of the model?

Leslie Tate Over a year ago

sorry for not specifying. sorting in descending order by average cost!

perl Over a year ago

Sure, you just add .nlargest(20), making it df.groupby('model')['cost'].mean().nlargest(20).plot.bar()

Leslie Tate Over a year ago

This makes sense. Thank you very much!

not_speshal · Accepted Answer · 2021-04-05 18:01:41Z

1

You can use the pandas plot function like so:

df.plot.bar(x='model', y='cost')

answered Apr 5, 2021 at 18:01

not_speshal

23.2k2 gold badges18 silver badges33 bronze badges

2 Comments

Leslie Tate Over a year ago

im struggling with plotting the "the average cost per model". How would i get the average/mean?

not_speshal Over a year ago

df.groupby('model')['cost'].mean()

Collectives™ on Stack Overflow

Matplotlib plotting dataframe

2 Answers 2

5 Comments

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

5 Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related