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

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

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()