Golang : Find correlation coefficient example
Below is an example of Golang code to find the strength of relation between two given variables or correlation coefficient. For financial analysis, it is useful in finding out if a variable has negative or positive correlation.
From investopedia.com:
A correlation of -1.0 shows a perfect negative correlation, while a correlation of 1.0 shows a perfect positive correlation. A correlation of 0.0 shows zero or no relationship between the movement of the two variables.
Here you go!
package main
import (
"fmt"
"math"
)
func main() {
X := []int{15, 18, 21, 24, 27}
Y := []int{25, 25, 27, 31, 32}
// Size of X array.
n := len(X)
// Get the strength of relation between two variables - X and 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++ {
// sum of elements of array X.
sum_X = sum_X + X[i]
// sum of elements of array Y.
sum_Y = sum_Y + Y[i]
// sum of X[i] * Y[i].
sum_XY = sum_XY + X[i]*Y[i]
// sum of square of array elements.
squareSum_X = squareSum_X + X[i]*X[i]
squareSum_Y = squareSum_Y + Y[i]*Y[i]
}
// use formula for calculating correlation
// coefficient.
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
}
Output:
0.9534625892455922
References :
https://www.investopedia.com/terms/c/correlationcoefficient.asp
https://www.geeksforgeeks.org/program-find-correlation-coefficient
https://www.socketloop.com/tutorials/golang-convert-cast-int-to-float-example
See also : Golang : Calculate Relative Strength Index(RSI) example
By Adam Ng(黃武俊)
IF you gain some knowledge or the information here solved your programming problem. Please consider donating to the less fortunate or some charities that you like. Apart from donation, planting trees, volunteering or reducing your carbon footprint will be great too.
Advertisement
Tutorials
+9.5k Swift : Convert (cast) String to Integer
+7k Golang : Multiplexer with net/http and map
+10.8k Golang : Determine if time variables have same calendar day
+9.2k Golang : Resolve domain name to IP4 and IP6 addresses.
+7.2k Golang : Number guessing game with user input verification example
+9.5k Golang : How to delete element(data) from map ?
+27.8k Golang : Get first few and last few characters from string
+7.9k Golang : Handle sub domain with Gin
+11.2k Golang : convert(cast) string to integer value
+5.2k Golang : Get missing location after unmarshal binary and gob decode time.
+3.9k Nginx and PageSpeed build from source CentOS example
+18.7k Golang : Accept input from user with fmt.Scanf skipped white spaces and how to fix it