3d plot linear regression python