溫馨提示×

使用Matplotlib如何繪制帶有回歸線的散點(diǎn)圖

小億
100
2024-05-20 12:59:31
欄目: 編程語言

要繪制帶有回歸線的散點(diǎn)圖，可以使用Matplotlib中的matplotlib.pyplot.scatter函數(shù)繪制散點(diǎn)圖，以及numpy.polyfit函數(shù)來擬合回歸線。

以下是一個(gè)示例代碼，演示如何使用Matplotlib繪制帶有回歸線的散點(diǎn)圖：

import numpy as np
import matplotlib.pyplot as plt

# 生成隨機(jī)數(shù)據(jù)
np.random.seed(0)
x = np.random.rand(100)
y = 2 * x + np.random.normal(0, 0.1, 100)

# 使用polyfit函數(shù)擬合回歸線
coefficients = np.polyfit(x, y, 1)  # 1表示一次多項(xiàng)式，即一條直線
poly = np.poly1d(coefficients)
trendline = poly(x)

# 繪制散點(diǎn)圖和回歸線
plt.scatter(x, y, label='Data points')
plt.plot(x, trendline, color='red', label='Regression line')
plt.xlabel('x')
plt.ylabel('y')
plt.legend()
plt.show()

運(yùn)行以上代碼將得到一個(gè)帶有回歸線的散點(diǎn)圖?？梢愿鶕?jù)需要調(diào)整數(shù)據(jù)生成的方式、回歸線的階數(shù)等來適應(yīng)不同的場景。

0 踩