Python Calculate R Squared Value Correlation