如何在 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

使用字符串格式：

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

产量