要在折线图中添加一条直线,可以使用 matplotlib 库中的 axhline 或 axvline 函数。这两个函数分别用于添加水平线和垂直线。
以下是一个简单的示例代码:
import matplotlib.pyplot as plt
import numpy as np
# 生成示例数据
x = np.arange(0, 10, 0.1)
y = np.sin(x)
# 绘制折线图
fig, ax = plt.subplots()
ax.plot(x, y)
# 添加一条水平线
ax.axhline(y=0.5, color='r', linestyle='--')
plt.show()
这段代码会生成一个包含折线图和一条水平线的图表。其中,axhline 函数的参数 y=0.5 表示添加的水平线的位置为 y=0.5,color='r' 表示线条颜色为红色,linestyle='--' 表示线条样式为虚线。
如果要添加一条垂直线,可以使用 axvline 函数,并设置参数 x 的值即可。