pandas dataframe binned data plotting

Question

I have a pandas dataframe (df) with a column df.age ranging from 0 - 100 years and a column df.haircolor (bright, dark, purple, grey).

Now I want to bin the age in decades:

bins = np.linspace(df.age.min(), df.age.max(), 10)
decades = df.groupby(np.digitize(df.age, bins))

Now I am trying to find a good way to plot this. I'd like a barplot where each haircolor has a bar. Naively I tried that.

df['haircolor'].plot(kind='bar', by=decades)

It is not giving me the result I hoped for. Anyone an idea? thanks.

Alexander · Accepted Answer · 2015-05-18 15:37:49Z

2

Try this:

df['decade'] = df.age // 10 * 10
df.groupby(['decade', 'haircolor']).haircolor.count().plot(kind='bar')

answered May 17, 2015 at 22:52

Alexander

111k32 gold badges212 silver badges208 bronze badges

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

4 Comments

That gives me a KeyError: 'decades'

Sorry, I thought you already had included the decade in your DataFrame. See above for calculation.

Thanks Alexander, that works - do you know a way to colorcode the different haircolors too?