python - 如何在 matplotlib (python) 中制作带有 r 值和回归函数的图例？

Question

以下代码:

import numpy as np
import matplotlib.pyplot as plt
from scipy import stats

x=np.linspace(2.0, 3.0, num=100)*1e-5
y=np.linspace(2.0, 3.0, num=100)*1e3

slope, intercept, r_value, p_value, std_err = stats.linregress(x,y)
predict_y = intercept + slope * x

plt.plot(x,y,'go')
plt.plot(x, predict_y, 'k-')
plt.legend(('data', 'line-regression'), 'upper left')
plt.autoscale(True)
plt.grid(True)
plt.show()

我可以绘制它，我可以找到 r 值，我可以进行线性回归并将其显示在图上。我傻到把它写在图表上。也许你能帮忙？

以便函数和 r 值显示为自己的图例或在图例中或专家认为最好的内容。

亲切的问候!提前联系!

附言!如果您知道更好的方法来解决我的问题，我也会很高兴:-)

Answer 1

使用string formatting :

plt.legend(('data', 'line-regression r={}'.format(r_value)), 'best')

产量