least-squares code snippets in python