добавить наклон и г ^ 2 в графике в python

Question

У меня есть этот код

plt.figure()
plt.subplot(2,2,1)
x1 = delta_ST_all.iloc[0:47,15].values.reshape(-1, 1)
y1 = delta_ST_all.iloc[0:47,1].values.reshape(-1, 1)  
linear_regressor = LinearRegression()
linear_regressor.fit(x1, y1) 
y1_pred = linear_regressor.predict(x1)
plt.scatter(x1, y1, label='AE 220-225m', s=20, color='lightpink')
plt.plot(x1, y1_pred, color='lightpink')

, но я хочу добавить легенду с уклоном и r ^ 2, может кто-нибудь помочь мне с этим? Я смог распечатать и увидеть его, но не могу найти код, чтобы поместить его в легенду на графике

Answer 1

По plt.text:

plt.text(0.8,0.9,
        'Slope={:.2g}\nR²{:.2g}'.format(float(slope),float(r2)),
        fontsize=16,ha='center')

, где slope, r2 - это значения, которые вы вычислили

Затем текст должен быть помещен в положение x = 0.8, y = 0,9, где x и y - значения ваших данных. Если вы хотите поместить текст в относительную позицию x, y к осям (что является единичным квадратом), то:

ax=plt.gca()
plt.text(0.8,0.9,
        'Slope={:.2g}\nR²{:.2g}'.format(float(slope),float(r2)),
        transform =ax.transAxes,fontsize=16,ha='center')

Здесь текст будет go в верхнем правом углу, где x = 0,8, у = 0,9 относительно осей.