要繪制帶有回歸線的散點圖,可以使用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()
運行以上代碼將得到一個帶有回歸線的散點圖。可以根據需要調整數據生成的方式、回歸線的階數等來適應不同的場景。