Least squares fit of a surface in Python (with ridiculous application)