Наложение линейной функции на точечную диаграмму ⇐ Python

[Anonymous]

Моя задача состоит в том, чтобы наложить график пользовательской линейной функции на уже имеющуюся диаграмму рассеяния. Код выглядит следующим образом:

Код: Выделить всё

base_beta = results.params
X_plot = np.linspace(0,1,400)

g = sns.FacetGrid(data, size = 6)
g = g.map(plt.scatter, "usable_area", "price", edgecolor="w")

Where

Код: Выделить всё

base_beta

is only a constant, and then one coefficient. Basically, I want to overlay a function that plots a line

Код: Выделить всё

y = constant + coefficient * x

I tried to overlay a line using this but it did not work.

Код: Выделить всё

g = g.map_dataframe(plt.plot, X_plot, X_plot*base_beta[1]+base_beta[0], 'r-')
plt.show()

The current scatter plot looks like so:



Can any one help me with this?

--ATTEMPT 1

Код: Выделить всё

base_beta = results.params
X_plot = np.linspace(0,1,400)
Y_plot = base_beta [0] + base_beta[1]*X_plot

g = sns.FacetGrid(data, size = 6)
g = g.map(plt.scatter, "usable_area", "price", edgecolor="w")
plt.plot(X_plot, Y_plot, color='r')
plt.show()

Resulted in the same graph but no line:




Источник: https://stackoverflow.com/questions/465 ... atter-plot