代码：
1 # -*- coding:utf-8 -*- 2 3 import numpy as np 4 import math 5 6 7 # 相关度 8 def computeCorrelation(X, Y): 9 xBar = np.mean(X) 10 yBar = np.mean(Y) 11 SSR = 0 12 varX = 0 13 varY = 0 14 for i in range(0, len(X)): 15 diffXXbar = X[i] - xBar 16 diffYYbar = Y[i] - yBar 17 SSR += (diffXXbar * diffYYbar) 18 varX += diffXXbar ** 2 19 varY += diffYYbar ** 2 20 SST = math.sqrt(varX * varY) 21 return SSR / SST 22 23 24 25 26 # print("相关度r:", computeCorrelation(testX, testY)) 27 # 相关度r: 0.940310076545 28 29 # R平方 30 # 简单线性回归： 31 # print("r^2:", str(computeCorrelation(testX, testY)**2)) 32 # r^2: 0.884183040052 33 34 # 多个x自变量时： 35 def polyfit(x, y, degree): # degree自变量x次数 36 result = {} 37 coeffs = np.polyfit(x, y, degree) 38 result['polynomial'] = coeffs.tolist() 39 40 p = np.poly1d(coeffs) 41 yhat = p(x) 42 ybar = np.sum(y) / len(y) 43 ssreg = np.sum((yhat - ybar) ** 2) 44 sstot = np.sum((y - ybar) ** 2) 45 result['determination'] = ssreg / sstot 46 47 return result 48 49 # 测试 50 testX = [1, 3, 8, 7, 9] 51 testY = [1,1,1,1,1] 52 53 inf=float("inf") 54 ninf=float("-inf") 55 nan=float("nan") 56 result=polyfit(testY, testX, 1)["determination"] 57 58 print(inf) 59 if result == inf: 60 print("PPPP") 61 # 测试 62 print(result) 63 # r^2:0.884183040052