Наложение числового значения медианы / дисперсии в boxplots

при использовании графиков в Python есть ли способ автоматически / легко наложить значение медианы и дисперсии поверх каждого окна (или, по крайней мере, числовое значение медианы)?

например. в boxplot ниже я хотел бы наложить текст (медиана, +- std) на каждом поле.

enter image description here

8
boxplot matplotlib python

1 ответов

предполагая, что вы используете функцию boxplot для рисования boxplots, она возвращает словарь, содержащий компоненты графика. Обратите внимание, что поле представляет внутренний диапазон квартилей (от 25 до 75 процентилей), а не стандартное отклонение.

>>> bp_dict = boxplot(data, vert=False) # draw horizontal boxplot
>>> bp_dict.keys()
>>> bp_dict.keys()
['medians', 'fliers', 'whiskers', 'boxes', 'caps']

Они содержат объекты Line2D, которые образуют каждый из элементов графика. Вы можете использовать Line2D.get_xydata метод, чтобы получить медиану и поле позиции (в координатах данных), чтобы выяснить, где расположить ваш текст.

from pylab import *

# from http://matplotlib.org/examples/pylab_examples/boxplot_demo.html

# fake up some data
spread= rand(50) * 100
center = ones(25) * 50
flier_high = rand(10) * 100 + 100
flier_low = rand(10) * -100
data =concatenate((spread, center, flier_high, flier_low), 0)

# fake up some more data
spread= rand(50) * 100
center = ones(25) * 40
flier_high = rand(10) * 100 + 100
flier_low = rand(10) * -100
d2 = concatenate( (spread, center, flier_high, flier_low), 0 )
data.shape = (-1, 1)
d2.shape = (-1, 1)
#data = concatenate( (data, d2), 1 )
# Making a 2-D array only works if all the columns are the
# same length.  If they are not, then use a list instead.
# This is actually more efficient because boxplot converts
# a 2-D array into a list of vectors internally anyway.
data = [data, d2, d2[::2,0]]

# multiple box plots on one figure
figure()

# get dictionary returned from boxplot
bp_dict = boxplot(data, vert=False)

for line in bp_dict['medians']:
    # get position data for median line
    x, y = line.get_xydata()[1] # top of median line
    # overlay median value
    text(x, y, '%.1f' % x,
         horizontalalignment='center') # draw above, centered

for line in bp_dict['boxes']:
    x, y = line.get_xydata()[0] # bottom of left line
    text(x,y, '%.1f' % x,
         horizontalalignment='center', # centered
         verticalalignment='top')      # below
    x, y = line.get_xydata()[3] # bottom of right line
    text(x,y, '%.1f' % x,
         horizontalalignment='center', # centered
             verticalalignment='top')      # below

show()

boxplot output

16