在 Python Matplotlib 中添加趋势线

Maxim Maeder 2023年1月30日 2022年5月11日
  1. 在 Matplotlib 中生成要绘制的数据
  2. 在 Python Matplotlib 中使用 NumPy 添加趋势线
在 Python Matplotlib 中添加趋势线

本教程将讨论在 Matplotlib 中向绘图添加趋势线。

在 Matplotlib 中生成要绘制的数据

在使用绘图之前,我们需要设置我们的脚本以使用该库。我们首先导入 Matplotlib。

我们从 random 模块加载 randrange 函数以快速生成一些数据。因此,请记住,这对你来说会有所不同。

import numpy
from matplotlib import pyplot as plt

x = [x for x in range(0, 10)]
y = numpy.random.rand(10)

在 Python Matplotlib 中使用 NumPy 添加趋势线

趋势线显示数据是增加还是减少。例如,地球上的整体温度可能看起来在波动,但它们正在上升。

我们用 NumPy 计算趋势线。为此,我们需要 x 轴和 y 轴。

然后我们使用 NumPypolyfitpoly1d 函数。最后,我们绘制趋势线。

# Plot the Data itself.
plt.plot(x, y)

# Calculate the Trendline
z = numpy.polyfit(x, y, 1)
p = numpy.poly1d(z)

# Display the Trendline
plt.plot(x, p(x))

输出:

在 Matplotlib 中添加趋势线

完整代码

import numpy
from matplotlib import pyplot as plt

x = [x for x in range(0, 10)]
y = numpy.random.rand(10)

# Plot the Data itself.
plt.plot(x, y)

# Calculate the Trendline
z = numpy.polyfit(x, y, 1)
p = numpy.poly1d(z)

# Display the Trendline
plt.plot(x, p(x))

plt.show()
Author: Maxim Maeder
Maxim Maeder avatar Maxim Maeder avatar

Hi, my name is Maxim Maeder, I am a young programming enthusiast looking to have fun coding and teaching you some things about programming.

GitHub