在 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