使用pyplot在plotlegend中显示fitparameters

Question

如果我对某些数据有所了解并且想要绘制，我也想要显示我得到的拟合参数以及我的情节图例中减少的卡方和r平方。 我需要做些什么才能完成这项工作？

很抱歉，如果这是一个微不足道的问题，我是python的新手，所以我可能不知道或看到明显的事情。

Answer 1

以下是如何执行拟合并将结果绘制在matplotlib中的图例中。

import numpy as np
import matplotlib.pyplot as plt
import scipy.optimize

x = np.arange(10)
y = np.arange(10)/5.+np.random.rand(10)

func = lambda x, a, b: a*x+b
popt, pcov = scipy.optimize.curve_fit(func, x,y)


plt.plot(x,y, marker="o", linestyle="none", label="data")

plt.plot(x,func(x, popt[0], popt[1]), label="fit"+str(popt))

plt.legend()
plt.show()