python - 如何在python中的散点图上绘制一条线？

Question

我有两个数据向量，我已将它们放入 pyplot.scatter()。现在我想对这些数据进行线性拟合。我该怎么做？我试过使用 scikitlearn 和 np.polyfit()。

Answer 1

import numpy as np
from numpy.polynomial.polynomial import polyfit
import matplotlib.pyplot as plt

# Sample data
x = np.arange(10)
y = 5 * x + 10

# Fit with polyfit
b, m = polyfit(x, y, 1)

plt.plot(x, y, '.')
plt.plot(x, b + m * x, '-')
plt.show()