皮尔逊相关系数 定义+python代码实现 （与王印讨论公式）

标签： python相关性系数皮尔逊系数

2016-10-25 20:59 153人阅读 评论(0) 收藏 举报

 分类：

集体智慧编程

版权声明：本文为博主原创文章，未经博主允许不得转载。

目录(?)[+]

皮尔逊相关系数

定义

皮尔逊相关系数判断两组数据与某一直线拟合程度的一种度量 
公式及其计算过程如下

 

python代码实现

'''@copyright Programming Collective Intelligence返回p1和p2的皮尔逊相关系数prefs是一个二维矩阵字典，如下图所示（名字，电影）  movie1   movie2       p1      4.5      5.0       p2      3.0      1.5 '''def simPearson(prefs, p1, p2):    #得到双方都曾评价过得物品列表    si={}    for item in prefs[p1]:        for item in prefs[p2]:            si[item] = 1    #得到列表元素的个数    n = len(si)    #如果两者没有共同之处，则返回1    if n == 0: return 1    #对所有偏好求和    sum1 = sum([prefs[p1][it] for it in si])    sum2 = sum([prefs[p2][it] for it in si])    #求平方和    sum1Sq = sum([pow(prefs[p1][it], 2) for it in si])    sum2Sq = sum([pow(prefs[p2][it], 2) for it in si])    #求乘积之和    pSum = sum([prefs[p1][it] * prefs[p2][it] for it in si])    #计算皮尔逊评价值    num = pSum - (sum1 * sum2) / n    den = sqrt((sum1Sq-pow(sum1, 2) / n) * (sum2Sq - pow(sum2, 2) / n))    if den == 0: return 0    r = num / den    return rhttp://blog.csdn.net/maxiaotiaoti/article/details/52927047  相关图书：《零基础学数据结构第二版》，机械工业出版社，2014。《CC++数据结构与算法速学速用大辞典》，中国铁道出版社