pandas plot with different variable for subplots and colour?

Question

Currently this code:

count_df = (df[['rank', 'name', 'variable', 'value']]
    .groupby(['rank', 'variable', 'name'])
    .agg('count')
    .unstack())
count_df .head()
#               value                          
# name           1lin STH_km27_lin ST_lin S_lin
# rank variable                                
# 1.0  NEE         24          115     33    28
#      Qg          23           54     14     9
#      Qh          37          124     11    28
# ...
count_df.plot(kind='bar')

gets me this plot:

using subplots=True in the .plot() call gets me this:

which is pretty useless, because the colours are mapped to the same variable as the subplot facetting. Is there a way to choose which column/index is used for the sub-plotting, so that I can still have colours per name (count_df column header), but sub-plots per variable, so that each subplot has a bar per name/rank, grouped by rank, and coloured by name?

naught101 · Accepted Answer · 2016-09-08 02:38:10Z

2

Hrm. I suspect this isn't doable in pandas by itself, but I found a way to do it in Seaborn:

import seaborn as sns

cdf = (df[['rank', 'name', 'variable', 'value']]
           .groupby(['rank', 'variable', 'name'])
           .agg('count'))
sns.factorplot(x="rank", y="value", row="variable", hue="name",
               data=cdf.reset_index(), kind='bar')

which results in:

which is close enough for my purposes

answered Sep 8, 2016 at 2:38

naught101

19.7k20 gold badges97 silver badges143 bronze badges

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

Collectives™ on Stack Overflow

pandas plot with different variable for subplots and colour?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related