Matplotlib different colors for bar graph based on value

user_8242299 0
Viewed 11242

I'm plotting returns for sectors and all the stocks in them. I would like to have the values > 100 to be green, and < 100 to be red. Here's my code:

sector_lst = ['XLK','XLF','XLE']  ## etc.

for i in sector_lst:                   
    fig = plt.figure(figsize=(12, 8)) 

    for x in sectordict[i]:                #sectordict[i] is all the stocks in a sector (so AAPL, GOOG, etc. for tech) 
        if pct_change[x] > 1:              
             pct_change[sectordict[i]].plot(kind='bar',color='g') 

        ##if pct_chg < 1
        else:                              
             pct_change[sectordict[i]].plot(kind='bar',color='r') 


plt.title(i)

So far this is returning the whole sector graphs as green or red; if the first value is > 100 all stocks will be green and vice versa.

My expected output is to have 11 graphs (which it currently does), but with different colors for each stock within the graph, if stock had > 100% return then it shows green and < 100% it shows red.

yahoo-finance pandas matplotlib python
2 Answers

You can achieve your result in a single call to plt.bar this way:

df = pd.DataFrame({'a' : [2,6,2,4,5,3,7]})
df['colors'] = 'r'
df.loc[df.a>=5,'colors'] = 'g'
plt.bar(df.index, df.a, color=df.colors)

You also mention in a comment that sectordict[i] is a dictionary, but you can easily convert a dictionary into a dataframe with: pd.DataFrame.from_dict.

user_27839430
Related