Julia中计算DataFrame两列数据相关系数的实现方法
Julia计算DataFrame两列相关系数的实现方法
Julia标准库的Statistics模块内置了cor函数,专门用于计算皮尔逊相关系数,实现逻辑和你给出的Python代码效果完全一致,且调用更简洁。
实现步骤
- 首先导入需要的依赖包:
DataFrames用于处理表格数据,Statistics为Julia标准库自带,无需额外安装 - 直接调用
cor函数传入两个目标列即可得到结果
代码示例
如果你的DataFrame变量名为data,直接执行以下代码即可得到相关系数:
using DataFrames, Statistics cor_score = cor(data.sensor1, data.sensor2)
完整可复现示例(对应你给出的样例数据)
using DataFrames, Statistics # 构造样例DataFrame data = DataFrame( datetime = ["2021-09-28 13:36:04", "2021-09-28 13:36:04", "2021-09-28 13:36:05", "2021-09-28 13:36:06"], sensor1 = [626, 622, 620, 619], sensor2 = [570, 571, 574, 578] ) # 计算相关系数 cor_score = cor(data.sensor1, data.sensor2)
该方法返回的是直接的相关系数数值,不需要像numpy的corrcoef那样额外取矩阵的[0,1]位置元素,调用更简便。
内容的提问来源于stack exchange,提问作者AfterFray
相关产品推荐
相关产品推荐

