python中打印曲面拟合公式

Question

我正在尝试使用matplotlib将曲面模型拟合到三维数据集（x、y、z）。
其中z = f(x,y)。
所以，我要用方程进行二次拟合：

f(x,y) = ax^2+by^2+cxy+dx+ey+f 

到目前为止，我已经成功地使用最小二乘法绘制了3d拟合曲面，使用的是：

^{pr2}$

但是，我怎样才能打印/得到曲面的拟合方程（带系数值）？在

我将非常感谢您的帮助。
谢谢您。在

Answer 1

根据功能文件scipy.linalg.lstsq公司http://docs.scipy.org/doc/scipy-0.15.1/reference/generated/scipy.linalg.lstsq.html估计的系数应该存储在变量C中（与A中的列相对应的顺序）。在

要打印带有估计系数的方程式，在小数点后显示2位数：

print 'f(x,y) = {:.2f}x^2+{:.2f}y^2+{:.2f}xy+{:.2f}x+{:.2f}y+{:.2f}'.format(C[4],C[5],C[3],C[1],‌​C[2],C[0])

或者：

^{pr2}$

顺便说一句，库pandas和{}对这类任务非常有用（例如检查Run an OLS regression with Pandas Data Frame）