Не могу нарисовать линейную регрессию на том же графике, что и диаграмма рассеяния - PullRequest
2 голосов
/ 23 февраля 2020

Я пытался построить линию линейной регрессии на моем графике рассеяния, но графики, кажется, показывают sh мой график рассеяния вниз. Диаграмма рассеяния выглядит нормально без линии регрессии. Пожалуйста, смотрите изображение 1.

Но как только я запускаю ax.plot(X,Y_pred,color= 'red') график искажается. Я проверил значения, они все кажутся в порядке. Я просто запутался, почему это происходит. Пожалуйста, помогите спасибо.

Изображение 1 enter image description here

Изображение 2 - С линией регрессии enter image description here

    #for regression line
X = latest_data1.iloc[:,3].values.reshape(-1,1)
Y= latest_data1.iloc[:,4].values.reshape(-1,1)
linear_regressor = LinearRegression()  # create object for the class
linear_regressor.fit(X,Y)
Y_pred = linear_regressor.predict(X)



fig, ax = plt.subplots()
ax.scatter(latest_data1['residential_land_area'],latest_data1['dwelling_units_under_management'])
ax.plot(X,Y_pred,color= 'red')
plt.title("Number of units vs Residential Land area in 2018 (Density)", fontsize=18, y=1.03)
plt.xlabel('Residential land area (hectares)')
plt.ylabel('Dwelling units')
ax.legend()
plt.show()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...