實現散點擬合直線的方法。
1. 導入所需庫umpyatplotlibumpyatplotlib庫則用于數據可視化。
portumpypportatplotlib.pyplot as plt
2. 生成數據umpydom模塊生成一些隨機數據。
pdomd(50)pdomd(50)
3. 繪制散點圖atplotlib庫繪制出散點圖。
plt.scatter(x,y)
plt.show()
4. 擬合直線umpy庫中的polyfit函數進行擬合。該函數可以返回擬合直線的斜率和截距。
p.polyfit(x, y, 1)
5. 繪制擬合直線atplotlib庫繪制出擬合直線。
plt.scatter(x,y)
plt.plot(x, kx + b, color='r')
plt.show()
6. 完整代碼
portumpypportatplotlib.pyplot as plt
生成數據pdomd(50)pdomd(50)
繪制散點圖
plt.scatter(x,y)
plt.show()
擬合直線p.polyfit(x, y, 1)
繪制擬合直線
plt.scatter(x,y)
plt.plot(x, kx + b, color='r')
plt.show()
通過以上代碼,我們可以生成一張散點圖并對其進行擬合直線,從而更加直觀地了解數據之間的關系。