Мне нужна линия, чтобы добраться до конечных углов графика, а также мне нужно изменить цвет линии на красный.Мой код:
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
Мне также нужно сделать то же самое для этого 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)
Неточный вывод для Scatterplot 2: Выход Scatterplot 2
Пример снимка экрана: Пример данных скриншота