How to display the value of the bar on each bar with pyplot.barh()?

后端 未结 9 660
不知归路
不知归路 2020-11-22 08:10

I generated a bar plot, how can I display the value of the bar on each bar?

Current plot:

\"enter

相关标签:
9条回答
  • 2020-11-22 08:23

    I know it's an old thread, but I landed here several times via Google and think no given answer is really satisfying yet. Try using one of the following functions:

    EDIT: As I'm getting some likes on this old thread, I wanna share an updated solution as well (basically putting my two previous functions together and automatically deciding whether it's a bar or hbar plot):

    def label_bars(ax, bars, text_format, **kwargs):
        """
        Attaches a label on every bar of a regular or horizontal bar chart
        """
        ys = [bar.get_y() for bar in bars]
        y_is_constant = all(y == ys[0] for y in ys)  # -> regular bar chart, since all all bars start on the same y level (0)
    
        if y_is_constant:
            _label_bar(ax, bars, text_format, **kwargs)
        else:
            _label_barh(ax, bars, text_format, **kwargs)
    
    
    def _label_bar(ax, bars, text_format, **kwargs):
        """
        Attach a text label to each bar displaying its y value
        """
        max_y_value = ax.get_ylim()[1]
        inside_distance = max_y_value * 0.05
        outside_distance = max_y_value * 0.01
    
        for bar in bars:
            text = text_format.format(bar.get_height())
            text_x = bar.get_x() + bar.get_width() / 2
    
            is_inside = bar.get_height() >= max_y_value * 0.15
            if is_inside:
                color = "white"
                text_y = bar.get_height() - inside_distance
            else:
                color = "black"
                text_y = bar.get_height() + outside_distance
    
            ax.text(text_x, text_y, text, ha='center', va='bottom', color=color, **kwargs)
    
    
    def _label_barh(ax, bars, text_format, **kwargs):
        """
        Attach a text label to each bar displaying its y value
        Note: label always outside. otherwise it's too hard to control as numbers can be very long
        """
        max_x_value = ax.get_xlim()[1]
        distance = max_x_value * 0.0025
    
        for bar in bars:
            text = text_format.format(bar.get_width())
    
            text_x = bar.get_width() + distance
            text_y = bar.get_y() + bar.get_height() / 2
    
            ax.text(text_x, text_y, text, va='center', **kwargs)
    

    Now you can use them for regular bar plots:

    fig, ax = plt.subplots((5, 5))
    bars = ax.bar(x_pos, values, width=0.5, align="center")
    value_format = "{:.1%}"  # displaying values as percentage with one fractional digit
    label_bars(ax, bars, value_format)
    

    or for horizontal bar plots:

    fig, ax = plt.subplots((5, 5))
    horizontal_bars = ax.barh(y_pos, values, width=0.5, align="center")
    value_format = "{:.1%}"  # displaying values as percentage with one fractional digit
    label_bars(ax, horizontal_bars, value_format)
    
    0 讨论(0)
  • 2020-11-22 08:38

    I was trying to do this with stacked plot bars. The code that worked for me was.

    # Code to plot. Notice the variable ax.
    ax = df.groupby('target').count().T.plot.bar(stacked=True, figsize=(10, 6))
    ax.legend(bbox_to_anchor=(1.1, 1.05))
    
    # Loop to add on each bar a tag in position
    for rect in ax.patches:
        height = rect.get_height()
        ypos = rect.get_y() + height/2
        ax.text(rect.get_x() + rect.get_width()/2., ypos,
                '%d' % int(height), ha='center', va='bottom')
    
    0 讨论(0)
  • 2020-11-22 08:41

    For anyone wanting to have their label at the base of their bars just divide v by the value of the label like this:

    for i, v in enumerate(labels):
        axes.text(i-.25, 
                  v/labels[i]+100, 
                  labels[i], 
                  fontsize=18, 
                  color=label_color_list[i])
    

    (note: I added 100 so it wasn't absolutely at the bottom)

    To get a result like this:

    0 讨论(0)
  • 2020-11-22 08:44

    Use plt.text() to put text in the plot.

    Example:

    import matplotlib.pyplot as plt
    N = 5
    menMeans = (20, 35, 30, 35, 27)
    ind = np.arange(N)
    
    #Creating a figure with some fig size
    fig, ax = plt.subplots(figsize = (10,5))
    ax.bar(ind,menMeans,width=0.4)
    #Now the trick is here.
    #plt.text() , you need to give (x,y) location , where you want to put the numbers,
    #So here index will give you x pos and data+1 will provide a little gap in y axis.
    for index,data in enumerate(menMeans):
        plt.text(x=index , y =data+1 , s=f"{data}" , fontdict=dict(fontsize=20))
    plt.tight_layout()
    plt.show()
    

    This will show the figure as:

    bar chart with values at the top

    0 讨论(0)
  • 2020-11-22 08:44

    For pandas people :

    ax = s.plot(kind='barh') # s is a Series (float) in [0,1]
    [ax.text(v, i, '{:.2f}%'.format(100*v)) for i, v in enumerate(s)];
    

    That's it. Alternatively, for those who prefer apply over looping with enumerate:

    it = iter(range(len(s)))
    s.apply(lambda x: ax.text(x, next(it),'{:.2f}%'.format(100*x)));
    

    Also, ax.patches will give you the bars that you would get with ax.bar(...). In case you want to apply the functions of @SaturnFromTitan or techniques of others.

    0 讨论(0)
  • 2020-11-22 08:47

    I have noticed api example code contains an example of barchart with the value of the bar displayed on each bar:

    """
    ========
    Barchart
    ========
    
    A bar plot with errorbars and height labels on individual bars
    """
    import numpy as np
    import matplotlib.pyplot as plt
    
    N = 5
    men_means = (20, 35, 30, 35, 27)
    men_std = (2, 3, 4, 1, 2)
    
    ind = np.arange(N)  # the x locations for the groups
    width = 0.35       # the width of the bars
    
    fig, ax = plt.subplots()
    rects1 = ax.bar(ind, men_means, width, color='r', yerr=men_std)
    
    women_means = (25, 32, 34, 20, 25)
    women_std = (3, 5, 2, 3, 3)
    rects2 = ax.bar(ind + width, women_means, width, color='y', yerr=women_std)
    
    # add some text for labels, title and axes ticks
    ax.set_ylabel('Scores')
    ax.set_title('Scores by group and gender')
    ax.set_xticks(ind + width / 2)
    ax.set_xticklabels(('G1', 'G2', 'G3', 'G4', 'G5'))
    
    ax.legend((rects1[0], rects2[0]), ('Men', 'Women'))
    
    
    def autolabel(rects):
        """
        Attach a text label above each bar displaying its height
        """
        for rect in rects:
            height = rect.get_height()
            ax.text(rect.get_x() + rect.get_width()/2., 1.05*height,
                    '%d' % int(height),
                    ha='center', va='bottom')
    
    autolabel(rects1)
    autolabel(rects2)
    
    plt.show()
    

    output:

    FYI What is the unit of height variable in "barh" of matplotlib? (as of now, there is no easy way to set a fixed height for each bar)

    0 讨论(0)
提交回复
热议问题