如何在matplotlib.hlines中设置标签？

前言

在使用matplotlib绘制图表时，我们经常需要标记出特殊点的坐标或者在某些地方添加注释信息。其中，使用hlines函数可以很方便地绘制水平线。但是，在绘制水平线的同时，如何在水平线上设置标签，成为一个问题。

本文将详细介绍如何在matplotlib的hlines函数中设置标签，帮助大家更好地进行图表绘制。

hlines函数简介

matplotlib.pyplot.hlines函数是绘制水平线的函数，其中常用参数如下：

matplotlib.pyplot.hlines(y, xmin, xmax, colors='k', linestyles='solid', label='', *, data=None, **kwargs)

y：水平线的y值；
xmin：水平线的起始x值；
xmax：水平线的结束x值；
color：水平线颜色，默认值为黑色；
linestyle：水平线的样式；
label：水平线的标签。

除了以上必备参数外，hlines函数还支持其他可选参数。例如，hlines函数中的**kwargs可以用来传递其他参数，这些参数将用于设置水平线的样式。下面是一个示例代码，展示了如何使用hlines函数绘制水平线。

import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots()

y = np.linspace(0, 10)
xmin = 2
xmax = 8

ax.hlines(y=y, xmin=xmin, xmax=xmax, colors='r', linestyles='dashed', label='sample line')

plt.legend(loc='upper right')
plt.show()

在hlines中添加标签

通过上面的示例我们可以看到，我们已经成功地绘制了一条水平的示例线，但是我们还没有在这条水平线上添加标签。为了添加标签，我们需要借助annotate函数。

annotate函数的具体参数可以参考官方文档，此处我们只介绍下面四个参数：

s: str
设置注释文本
xy: tuple
设置注释文本的位置，注释箭头终点位置
xytext: tuple
设置注释文本的起始位置，注释箭头起始位置
arrowprops: dict
设置注释箭头的属性

下面的代码演示了如何在水平线上添加标签。

fig, ax = plt.subplots()

y = np.linspace(0, 10)
xmin = 2
xmax = 8

ax.hlines(y=y, xmin=xmin, xmax=xmax, colors='r', linestyles='dashed')

ax.annotate('sample line', xy=(xmin, y[5]), xytext=(xmin-1, y[5]),
            arrowprops=dict(facecolor='black', headlength=10, width=3))

plt.show()

多条水平线上添加标签

如果需要在一个图中添加多条水平线，并且需要在每条线上添加标签，我们可以沿用上面的方法，为每条水平线都单独添加一个标签。例如下面的代码：

fig, ax = plt.subplots()

y = np.linspace(0, 10)
xmin1 = 2
xmax1 = 8
xmin2 = 3
xmax2 = 7

ax.hlines(y=y, xmin=xmin1, xmax=xmax1, colors='r', linestyles='dashed')
ax.annotate('sample line1', xy=(xmin1, y[5]), xytext=(xmin1-1, y[5]),
arrowprops=dict(facecolor='black', headlength=10, width=3))

ax.hlines(y=y, xmin=xmin2, xmax=xmax2, colors='b', linestyles='dashed')
ax.annotate('sample line2', xy=(xmin2, y[3]), xytext=(xmin2-1, y[3]),
            arrowprops=dict(facecolor='black', headlength=10, width=3))

plt.show()