js计算皮尔逊相关系数

代码如下;

复制代码
let XGX= {
    correlationCoefficient(pA, pB) {
        let covXY = -pA * pB
        let varX = pA * (1-pA) 
        let varY = (1-pB)* pB
        let res = covXY / (Math.sqrt(varX*varY, 2))
        return res
    },
    correlation(x,y){
        x=[0.3,50.2,99.5,199.3,299,398];
        y=[0.1,50,99.9,200,300,400];
        // 计算均值
        const avgX=x.reduce((a,b)=>a+b,0)/x.length;
        const avgY=y.reduce((a,b)=>a+b,0)/y.length;
        //计算差值
        const diffX=x.map((value,index)=>value-avgX);
        const diffY=y.map((value,index)=>value-avgY);
        //计算乘积
        const product =diffX.map((value,index)=>value*diffY[index]);
        //计算相关系数
        const sumProduct=product.reduce((a,b)=>a+b,0);
        const correlation=sumProduct/(x.length*y.length);
        return correlation
    },
    /**
     * 计算皮尔逊相关系数
     * @list1:数字列表1
     * @list2:数字列表2
     * @return number
    * */
    pearson(list1,list2){
        let newList = [];
        let len = list1.length;
        for(let i=0;i<len;i++){newList.push(list1[i]*list2[i]);}
        let mean1 = this.mean(list1),mean2=this.mean(list2);
        let cov = this.mean(newList)-mean1*mean2;
        return cov/(this.std(list1,mean1)*this.std(list2,mean2))
    },
    /**
     * 计算平均值
     * @list:数字列表
     * @return number
     * */
    mean(list){
        let sum = 0;
        list.forEach((item)=>{sum+=item})
        return sum/list.length;
    },
    /**
     * 计算标准差
     * @list:数字列表
     * @mean:列表list的平均值
     * @return number
     * */
    std(list,mean){
        let sum =0;
        list.forEach((item)=>{sum+=Math.pow(item-mean,2)})
        return Math.sqrt(sum/list.length)
    }
};
export default XGX

调用:

效果:

相关推荐
Eiceblue35 分钟前
【免费.NET方案】CSV到PDF与DataTable的快速转换
开发语言·pdf·c#·.net
m0_555762901 小时前
Matlab 频谱分析 (Spectral Analysis)
开发语言·matlab
像风一样自由20201 小时前
HTML与JavaScript:构建动态交互式Web页面的基石
前端·javascript·html
浪裡遊2 小时前
React Hooks全面解析:从基础到高级的实用指南
开发语言·前端·javascript·react.js·node.js·ecmascript·php
lzb_kkk3 小时前
【C++】C++四种类型转换操作符详解
开发语言·c++·windows·1024程序员节
好开心啊没烦恼3 小时前
Python 数据分析:numpy,说人话,说说数组维度。听故事学知识点怎么这么容易?
开发语言·人工智能·python·数据挖掘·数据分析·numpy
简佐义的博客4 小时前
破解非模式物种GO/KEGG注释难题
开发语言·数据库·后端·oracle·golang
Liudef064 小时前
2048小游戏实现
javascript·css·css3
程序员爱钓鱼4 小时前
【无标题】Go语言中的反射机制 — 元编程技巧与注意事项
开发语言·qt