Как отобразить значение полосы на каждой панели с помощью pyplot.barh ()?

107

Я создал столбчатый график, как я могу отобразить значение столбца на каждом столбце?

Текущий сюжет:

введите описание изображения здесь

Что я пытаюсь получить:

введите описание изображения здесь

Мой код:

import os
import numpy as np
import matplotlib.pyplot as plt

x = [u'INFO', u'CUISINE', u'TYPE_OF_PLACE', u'DRINK', u'PLACE', u'MEAL_TIME', u'DISH', u'NEIGHBOURHOOD']
y = [160, 167, 137, 18, 120, 36, 155, 130]

fig, ax = plt.subplots()    
width = 0.75 # the width of the bars 
ind = np.arange(len(y))  # the x locations for the groups
ax.barh(ind, y, width, color="blue")
ax.set_yticks(ind+width/2)
ax.set_yticklabels(x, minor=False)
plt.title('title')
plt.xlabel('x')
plt.ylabel('y')      
#plt.show()
plt.savefig(os.path.join('test.png'), dpi=300, format='png', bbox_inches='tight') # use format='svg' or 'pdf' for vectorial pictures
Франк Дернонкур
источник

Ответы:

175

Добавить:

for i, v in enumerate(y):
    ax.text(v + 3, i + .25, str(v), color='blue', fontweight='bold')

результат:

введите описание изображения здесь

Y-значения vявляются как ax.textкоординатами x, так и строковыми значениями , и для удобства столбчатая диаграмма имеет метрику 1 для каждого столбца, поэтому перечисление iявляется координатой y.

cphlewis
источник
11
возможно, замените use va = 'center' вместо "i + .25" для горизонтального выравнивания
Матхауз
13
plt.text(v, i, " "+str(v), color='blue', va='center', fontweight='bold')
Жоао Картушо
Как я могу подавить вывод текста в записной книжке jupyter? ";" в конце не работает.
Ральф Хундевадт
Что нужно сделать, если я хочу напечатать NEIGHBORHOOD в верхней части панели и то же самое для всех полос baove, а не слева. пробовал искать во многих местах, но до сих пор не знаю.
Риши
@RalfHundewadt просто поставил plt.show()пункт в конце. Например: df.plot(); plt.show()
Хайро Алвес,
36

Я заметил, что код примера api содержит пример гистограммы со значением столбца, отображаемым на каждом столбце:

"""
========
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()

вывод:

введите описание изображения здесь

FYI Какая единица измерения переменной высоты в «barh» matplotlib? (на данный момент нет простого способа установить фиксированную высоту для каждой полосы)

Франк Дернонкур
источник
1
Похоже, что get_x () округляет число в большую сторону, даже если исходный x имеет десятичные знаки. Как сделать так, чтобы отображалось больше десятичных знаков?
ru111
1
опаздывает на вечеринку, но для всех, кто использует это, использование высоты + 0,1 вместо 1,05 * высота в функции автоматической метки создает постоянный разрыв между полосой и текстом
jnPy
19

Для тех, кто хочет иметь свою метку в основании своих столбцов, просто разделите v на значение метки следующим образом:

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

(примечание: я добавил 100, так что это было не совсем внизу)

Чтобы получить такой результат: введите описание изображения здесь

Скромак
источник
Неужто v == labels[i]и так могла просто быть третья и четвертая строчки 101, v?
НЕ.
15

Используйте plt.text (), чтобы вставить текст в график.

Пример:

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()

Это покажет цифру как:

гистограмма со значениями вверху

Анирван Сен
источник
В любом случае сдвинуть текст немного влево?
Рамджит,
1
@ S.Ramjitplt.text(x=index , y =data+1 , s=f"{data}" , fontdict=dict(fontsize=20), va='center')
theGtknerd
14

Я знаю, что это старая ветка, но я несколько раз заходил сюда через Google и думаю, что ни один из полученных ответов пока не удовлетворяет. Попробуйте использовать одну из следующих функций:

РЕДАКТИРОВАТЬ : Поскольку я получаю несколько лайков в этой старой ветке, я также хочу поделиться обновленным решением (в основном, объединяя две мои предыдущие функции и автоматически решая, является ли это графиком bar или hbar):

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)

Теперь вы можете использовать их для обычных гистограмм:

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)

или для горизонтальных полос:

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)
СатурнОтТитана
источник
10

Для людей панд:

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)];

Вот и все. В качестве альтернативы для тех, кто предпочитает applyповторение цикла с помощью enumerate:

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

Кроме того, ax.patchesвы получите бары, которые вы получите ax.bar(...). Если вы хотите применить функции @SaturnFromTitan или чужие техники.

tozCSS
источник
Убедитесь, что i, v не перевернуты. Может так и должно быть[ax.text(i, v, '{:.2f}%'.format(100*v)) for i, v in enumerate(s)];
Хайро Алвес
@JairoAlves - это горизонтальная полоса, а v представляет положение на оси x, поэтому оно должно быть правильным. См. Также принятый ответ.
tozCSS
Пример с патчами for p in ax.patches: ax.annotate(str(p.get_height()), (p.get_x() * 1.005, p.get_height() * 1.005))
Ichta
1

Мне также нужны были ярлыки полос, обратите внимание, что моя ось Y имеет увеличенный вид с использованием ограничений по оси Y. Расчеты по умолчанию для размещения меток поверх панели все еще работают с использованием высоты (use_global_coordinate = False в примере). Но я хотел показать, что метки можно размещать и внизу графика в увеличенном виде, используя глобальные координаты в matplotlib 3.0.2 . Надеюсь, это кому-то поможет.

def autolabel(rects,data):
"""
Attach a text label above each bar displaying its height
"""
c = 0
initial = 0.091
offset = 0.205
use_global_coordinate = True

if use_global_coordinate:
    for i in data:        
        ax.text(initial+offset*c, 0.05, str(i), horizontalalignment='center',
                verticalalignment='center', transform=ax.transAxes,fontsize=8)
        c=c+1
else:
    for rect,i in zip(rects,data):
        height = rect.get_height()
        ax.text(rect.get_x() + rect.get_width()/2., height,str(i),ha='center', va='bottom')

Пример вывода

Харамоз
источник
0

Я пытался сделать это с помощью составных столбцов графика. Код, который работал у меня, был.

# 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

Проверьте эту ссылку Галерея Matplotlib Вот как я использовал фрагмент кода autolabel.

    def autolabel(rects):
    """Attach a text label above each bar in *rects*, displaying its height."""
    for rect in rects:
        height = rect.get_height()
        ax.annotate('{}'.format(height),
                    xy=(rect.get_x() + rect.get_width() / 2, height),
                    xytext=(0, 3),  # 3 points vertical offset
                    textcoords="offset points",
                    ha='center', va='bottom')
        
temp = df_launch.groupby(['yr_mt','year','month'])['subs_trend'].agg(subs_count='sum').sort_values(['year','month']).reset_index()
_, ax = plt.subplots(1,1, figsize=(30,10))
bar = ax.bar(height=temp['subs_count'],x=temp['yr_mt'] ,color ='g')
autolabel(bar)

ax.set_title('Monthly Change in Subscribers from Launch Date')
ax.set_ylabel('Subscriber Count Change')
ax.set_xlabel('Time')
plt.show()
Партия Пандей
источник