bar plot with different colors in python

Question

I have data like this :

import pandas as pd
import matplotlib.pyplot as plt
index={'A','B','C','D','E'}
d={'typ':[1,2,2,2,1],'value':[10,25,15,17,13]}
df=pd.DataFrame(d,index=index)

I want to plot the dataframe in horizontal bars with different colors reffering to the column 'typ'

Perhaps the accepted answer to another matplotlib question may help you: it cycles through the individual bars (patches) and assigns a color to each patch individually. You'll need to adopt it for a barchart plot. — user707650
– user707650, Commented Jul 18, 2015 at 11:33

Marijn van Vliet · Accepted Answer · 2016-06-03 06:11:12Z

5

You can use the color parameter of matplotlib's barh function:

import pandas as pd
import matplotlib.pyplot as plt
index={'A','B','C','D','E'}
d={'typ':[1,2,2,2,1],'value':[10,25,15,17,13]}
df=pd.DataFrame(d,index=index)

# define the colors for each type
colors = {1:'blue', 2:'red'}

# plot the bars
plt.bar(range(len(df)), df['value'], align='center',
        color=[colors[t] for t in df['typ']])

The resulting figure

edited Jun 3, 2016 at 6:11

answered Jul 19, 2015 at 18:48

Marijn van Vliet

5,4092 gold badges36 silver badges46 bronze badges

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

Collectives™ on Stack Overflow

bar plot with different colors in python

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related