Создание обозначенных горизонтальных линий на графике - PullRequest
0 голосов
/ 09 июля 2019

Я пытаюсь воспроизвести эту диаграмму:

enter image description here

но у меня проблемы с созданием горизонтальных линий с помощью столбцов. Я пробовал annotate и hlines, но они не дают того эффекта, который мне нужен. импорт matplotlib.pyplot как plt

plt.grid(which = 'both')
plt.xticks(fontsize = 16)
plt.yticks(fontsize = 16)
plt.xlim(-0.5,8)
plt.ylim(-0.5,10)
plt.xlabel('Redshift, z', fontsize = 16)
plt.hlines(8, 0, .3)
plt.annotate(r'H$\alpha$', fontsize = 16, xy = (0,8), xycoords='data', xytext=(0,8), textcoords='data',
             arrowprops=dict(arrowstyle='<|-|>', connectionstyle='arc3', color = 'k', lw=2))
fig = plt.gcf()
width, height = 15,35   #   inches
fig.set_size_inches(width, height, forward = True)
plt.show()

Каков наилучший способ производства таких баров?

Ответы [ 2 ]

0 голосов
/ 09 июля 2019

Принятый ответ работает отлично, спасибо.

Кроме того, я автоматизировал цвета следующим образом:

colors = iter(cm.tab10(np.linspace(0,0.8,13)))

colour = 'k'
for y,xs,t in zip(bar_ys,bar_xs,bar_texts):
    plt.annotate('', xy = (xs[0],y), xycoords='data', xytext=(xs[1],y),
                 arrowprops=dict(arrowstyle='|-|', color=colour, lw=2, shrinkA=0, shrinkB=0))
    plt.annotate(t, xy = (xs[1],y), xycoords='data', xytext=(-5,5), textcoords='offset points',
                 fontsize = 16, va='baseline', ha='right', color=colour)
    colour = next(colors)
0 голосов
/ 09 июля 2019

Я бы использовал annotate напрямую, но для большей гибкости я бы отделил чертеж горизонтальных полос и соответствующий текст

plt.figure()
plt.grid(which = 'both')
plt.xticks(fontsize = 16)
plt.yticks(fontsize = 16)
plt.xlim(-0.5,8)
plt.ylim(-0.5,10)
plt.xlabel('Redshift, z', fontsize = 16)

bar_ys = [8,4]
bar_xs = [[0,6],[3,5]]
bar_texts = [r'H$\alpha$',r'H$\beta$']
bar_color = ['k','orange']

for y,xs,t,c in zip(bar_ys,bar_xs,bar_texts,bar_color):
    plt.annotate('', xy = (xs[0],y), xycoords='data', xytext=(xs[1],y),
                 arrowprops=dict(arrowstyle='|-|', color=c, lw=2, shrinkA=0, shrinkB=0))
    plt.annotate(t, xy = (xs[1],y), xycoords='data', xytext=(-5,5), textcoords='offset points',
                 fontsize = 16, va='baseline', ha='right', color=c)
plt.show()

enter image description here

...