I am trying to map a surface model to a 3D dataset (x, y, z) using matplotlib.
Where z = f(x,y).
So I'm going for a quadratic fitting with the equation:
f(x,y) = ax^2+by^2+cxy+dx+ey+f
So far, I have successfully built a 3D surface using the least squares method using:
A = np.c_[np.ones(data.shape[0]), data[:,:2], np.prod(data[:,:2], axis=1), data[:,:2]**2]
C,_,_,_ = scipy.linalg.lstsq(A, data[:,2])
Z = np.dot(np.c_[np.ones(XX.shape), XX, YY, XX*YY, XX**2, YY**2], C).reshape(X.shape)
But, how can I print / get the set surface equation (with coefficients)?
It will be of little use to me very much. thank.