如何在 matplotlib 的子图上添加垂直线

在 matplotlib 中，我们经常需要在子图上添加一些额外的辅助线来提高可视化质量，其中垂直线是常用的一种类型。下面介绍几种常见的在子图上添加垂直线的方式。

使用 axvline()

axvline() 是 matplotlib 中专门用来添加垂直线的函数，可以在指定横坐标位置上添加垂直线。使用方法如下：

import matplotlib.pyplot as plt

fig, ax = plt.subplots()

# 添加垂直线
ax.axvline(x=0.5, color='k', linestyle='--')

plt.show()

其中参数 x 指定垂直线的位置， color 和 linestyle 分别指定了垂直线的颜色和线型。

使用 vlines()

vlines() 是另一种添加垂直线的方式，它和 axvline() 的用途类似，不同的是 vlines() 可以同时添加多条垂直线。使用方法如下：

import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots()

# 添加多条垂直线
x = np.arange(-1, 1.1, 0.2)
y = np.sin(x)
ax.vlines(x=x, ymin=0, ymax=y, color='r', linestyle='--')

plt.show()

其中参数 x 指定垂直线的位置， ymin 和 ymax 分别指定垂直线的起始和终止位置，color 和 linestyle 分别指定了垂直线的颜色和线型。

使用 axhline()

最后，我们还介绍一种添加水平线的方法，因为很多时候垂直线和水平线会同时出现在一个子图中。 axhline() 是 matplotlib 中专门用来添加水平线的函数。使用方法如下：

import matplotlib.pyplot as plt

fig, ax = plt.subplots()

# 添加水平线
ax.axhline(y=0.5, color='k', linestyle='--')

plt.show()

其中参数 y 指定水平线的位置， color 和 linestyle 分别指定了水平线的颜色和线型。

上述三种方法可以结合起来，以实现更复杂的辅助线需求。