如何格式化散点图上的线

Question

我需要线条到达图形的末端，并且还需要将线条的颜色更改为红色。我的代码：

plt.figure(),
plt.subplot(211) 
plt.scatter(df.NOX, df.PTRATIO)
b, m = polyfit(x, y, 1)
plt.plot(x, y, '.' )
plt.plot(x, b + m * x, '-' )
plt.show()

Scatterplot

我还需要对此散点图执行相同的操作。 Scatterplot2 我使用的代码：

X_prime = np.linspace(df.PTRATIO.min(), df.PTRATIO.max(), 100)
X_prime = sm.add_constant(X_prime)
y_hat = lr_model1.predict(X_prime)

plt.figure()
plt.subplot(211) 
plt.scatter(df.NOX, df.PTRATIO)

plt.subplot(212)
plt.scatter(X.NOX, y)
plt.plot(X_prime[:,1], y_hat, 'red', alpha=0.9)

散点图2的输出不正确： Scatterplot 2 output

示例数据屏幕截图：Sample Data screenshot