溫馨提示×

使用Matplotlib如何繪制帶有回歸線的散點圖

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

要繪制帶有回歸線的散點圖,可以使用Matplotlib中的matplotlib.pyplot.scatter函數繪制散點圖,以及numpy.polyfit函數來擬合回歸線。

以下是一個示例代碼,演示如何使用Matplotlib繪制帶有回歸線的散點圖:

import numpy as np
import matplotlib.pyplot as plt

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

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

# 繪制散點圖和回歸線
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()

運行以上代碼將得到一個帶有回歸線的散點圖??梢愿鶕枰{整數據生成的方式、回歸線的階數等來適應不同的場景。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女