Python Get Correlation Coefficient Regression