添加趋势线以绘制 matplotlib - Python

在数据可视化中，趋势线是一种可视化工具，可帮助我们识别和预测数据的趋势和模式。在 Python 中，matplotlib 库提供了一种轻松添加趋势线的方法。

准备工作

在开始之前，我们需要确保已经安装了 matplotlib 库。如果还没有安装，可以使用以下命令进行安装：

!pip install matplotlib

接下来，让我们导入必要的库，并创建一些示例数据，以便演示如何绘制趋势线。

import matplotlib.pyplot as plt
import numpy as np

# 创建示例数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([5, 7, 9, 11, 13])

绘制折线图

首先，我们需要绘制一个基本的折线图，这可以使用 plot() 函数轻松完成。

plt.plot(x, y)
plt.show()

这将绘制出以下折线图：

添加趋势线

现在，我们将添加一个趋势线。这可以使用 polyfit() 函数计算出线性趋势线的参数。然后，我们可以使用这些参数创建趋势线的公式，并使用 plot() 函数进行绘制。

# 计算趋势线的参数
slope, intercept = np.polyfit(x, y, 1)

# 创建趋势线的公式
trendline = slope * x + intercept

# 绘制折线图和趋势线
plt.plot(x, y)
plt.plot(x, trendline, color='red', linestyle='--')
plt.show()

这将绘制出以下带有趋势线的折线图：

趋势线用红色虚线表示。

我们可以通过更改线条的颜色，样式和宽度来自定义趋势线。另外，如果要绘制非线性趋势线，则可以使用 polyfit() 函数的高阶参数。这些高阶参数将导致趋势线变得更曲线。