在 Python Matplotlib 中新增趨勢線
Maxim Maeder
2023年1月30日
2022年5月11日
本教程將討論在 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 軸。
然後我們使用 NumPy
的 polyfit
和 poly1d
函式。最後,我們繪製趨勢線。
# 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))
輸出:
完整程式碼
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
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