How to plot such a graph with matplotlib

Question

I have a dataframe:

  date       date2      col1   col2
 
2011-05-01  2011-05-01    3     25
            2011-05-02    1     17
            2011-05-03    2     12

2012-05-01  2012-05-01    5     45
            2012-05-02    5     73

How can I plot each column (col1 and col2) grouping dataframe by level 0 (date)?

I want to get a graph like this one in the picture. So date is determined by color, date2 is on x-axis and col1 or col2 are the cols I what to plot separately.

I tried stackplot, but it didn't work, as I can't set the x length:

plt.stackplot(df.date2,df.groupby(level=0)['col1'].sum())

Picture I want to receive:

Thank you

Try this: df.unstack('date2')['col1'].plot.area().. not enough data to duplicate your plot. — Scott Boston
– Scott Boston, Commented Sep 28, 2020 at 19:20
@user12628549 Please consider accepting and upvoting the answer if works for you. — Grayrigel
– Grayrigel, Commented Sep 28, 2020 at 19:28

Scott Boston · Accepted Answer · 2020-09-28 20:07:26Z

2

Try this using dataframe reshaping:

df.unstack('date2')['col1'].plot.area()

As @QuangHoang states, this is less costly, you are manipulating fewer columns, this is better:

df['col1'].unstack('date2').plot.area()

edited Sep 28, 2020 at 20:07

answered Sep 28, 2020 at 19:24

Scott Boston

154k15 gold badges160 silver badges207 bronze badges

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

2 Comments

Quang Hoang Over a year ago

Upvoted. df['col1'].unstack('date2').plot.area() is a little less costly.

Scott Boston Over a year ago

Thanks, @QuangHoang. Be safe and stay healthy.

Collectives™ on Stack Overflow

How to plot such a graph with matplotlib

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related