python - 在 plot python 中编写回归方程

Question

我做了一个回归，所以我得到：

slope_b = -0.7583790156301706
intercept_b = 7.22190409922338
r_value_b = -0.887981466827434

我想将回归方程添加到图中，如：

y = -0.76x + 7.22
r = -0.88

由于我找不到方法，所以我确实喜欢了一条文字：

axarr[0].text(8,6,'y=-0.76x+7.22')
axarr[0].text(9,5, 'r²=-0.88')

但是，我一直在回归中添加变量，所以斜率、截距和 r_value 都在变化，所以我每次都必须更改表达式。我做了一些研究，我发现了这个：

axarr[0].text(8,6,'y=', slope_b, '*x+', intercept_b)

但我没有得到想要的结果。另外，有没有办法让这些值只有两位小数？我更喜欢等式中的一个较短的数字，而不是这个：-0.7583790156301706。

标签： pythonplotequation

Answer 1

我刚刚解决了它：

axarr[0].plot(x,y,'r', label="y={0:.2f}x+{1:.2f}\nr{2:.2f}".format(slope_b,intercept_b,r_value_b), color='black')
axarr[0].legend()

其中 {0:.2f} 使数字具有两位小数