在go语言中实现查找相关系数示例代码

Go  /  管理员 发布于 1年前   283

在go语言中实现查找两个给定变量之间的关系强度或相关系数，比如：在财务分析中，它有助于找出变量之间是负相关还是正相关。

摘自 investopedia.com：

https://www.investopedia.com/terms/c/correlationcoefficient.asp

相关系数为-1.0 表示完全负相关，而相关系数为 1.0 表示完全正相关。

相关性为 0.0 表示两个变量的变动之间没有关系。

示例代码：

package main

 import (
     "fmt"
     "math"
 )
 
 func main() {
     X := []int{15, 18, 21, 24, 27}
     Y := []int{25, 25, 27, 31, 32}
     
     // X 数组的大小
     n := len(X)
     // 获取 X 和 Y 这两个变量之间的关系强度
     fmt.Println(correlationCoefficient(X, Y, n))
  }
     
  func correlationCoefficient(X []int, Y []int, n int) float64 {
     sum_X := 0
     sum_Y := 0
     sum_XY := 0
     squareSum_X := 0
     squareSum_Y := 0
     
     for i := 0; i < n; i++ {
         // 数组 X 的元素之和
         sum_X = sum_X + X[i]
         
         // 数组 Y 的元素之和
         sum_Y = sum_Y + Y[i]
         
         // X[i] * Y[i] 之和
         sum_XY = sum_XY + X[i]*Y[i]
         
         // 数组元素的平方和
         squareSum_X = squareSum_X + X[i]*X[i]
         squareSum_Y = squareSum_Y + Y[i]*Y[i]
     }
     
     // 使用公式计算相关
     // 系数
     corr := float64((n*sum_XY - sum_X*sum_Y)) /
     (math.Sqrt(float64((n*squareSum_X - sum_X*sum_X) * (n*squareSum_Y - sum_Y*sum_Y))))
     
     return corr
 }

输出：

0.9534625892455922