R中是否存在类似cor()的函数,可计算变量间成对真阳性值矩阵?
解决方案
你不需要额外安装第三方包,用R基础的矩阵乘法就可以快速得到你要的真阳性计数矩阵,原理是0-1矩阵的转置和自身相乘后,得到的矩阵元素就是对应两列同时为1的次数,和你要的计算规则完全匹配,计算效率远高于循环遍历方案。
示例代码
首先构造你提供的测试数据集:
df <- data.frame( A = c(1, 0, 1, 1), B = c(0, 0, 1, 0), C = c(0, 1, 1, 1), D = c(1, 1, 1, 1) )
直接运行如下代码即可得到目标矩阵:
tp_matrix <- t(df) %*% as.matrix(df)
如果你想要像cor()一样直接调用的函数,可以自己简单封装:
# 自定义和cor用法一致的真阳性计数函数 tp_count <- function(data) { # 可选:校验输入是否为0-1变量 if (!all(data %in% c(0, 1))) { warning("输入数据包含非0/1值,计算结果可能不符合预期") } data_matrix <- as.matrix(data) return(t(data_matrix) %*% data_matrix) } # 调用方式和cor完全一致 tp_count(df)
内容的提问来源于stack exchange,提问作者bms2202
相关产品推荐
相关产品推荐

