Implementing Statistical Functions: Mean, Median, Mode in Go

Understanding Statistical Functions in Go

Statistical functions are essential tools in data processing and analysis. They help summarize data and derive meaningful insights. This article will guide you through implementing three fundamental statistical functions: Mean, Median, and Mode in Go.

Mean

The mean, often referred to as the average, is calculated by summing all the elements in a dataset and dividing by the number of elements.

// Basic implementation of Mean in Go
package main

import "fmt"

func mean(numbers []float64) float64 {
    sum := 0.0
    for _, number := range numbers {
        sum += number
    }
    return sum / float64(len(numbers))
}

func main() {
    data := []float64{1.0, 2.0, 3.0, 4.0, 5.0}
    fmt.Printf("Mean: %.2f\n", mean(data))
}

Median

The median is the middle value of a dataset when sorted in ascending order. If the dataset has an even number of observations, the median is the average of the two middle numbers.

// Intermediate implementation of Median in Go
package main

import (
    "fmt"
    "sort"
)

func median(numbers []float64) float64 {
    sort.Float64s(numbers)
    len := len(numbers)
    if len%2 == 0 {
        return (numbers[len/2-1] + numbers[len/2]) / 2
    }
    return numbers[len/2]
}

func main() {
    data := []float64{3.0, 5.0, 1.0, 4.0, 2.0}
    fmt.Printf("Median: %.2f\n", median(data))
}

Mode

The mode is the number that appears most frequently in a dataset. A dataset may have one mode, more than one mode, or no mode at all.

// Advanced implementation of Mode in Go
package main

import "fmt"

func mode(numbers []int) []int {
    frequencyMap := make(map[int]int)
    maxFrequency := 0
    var modes []int

    for _, number := range numbers {
        frequencyMap[number]++
        if frequencyMap[number] > maxFrequency {
            maxFrequency = frequencyMap[number]
        }
    }

    for number, frequency := range frequencyMap {
        if frequency == maxFrequency {
            modes = append(modes, number)
        }
    }

    return modes
}

func main() {
    data := []int{1, 2, 2, 3, 3, 3, 4}
    fmt.Printf("Mode: %v\n", mode(data))
}

Conclusion

Implementing these statistical functions in Go provides a deeper understanding of data manipulation and statistical analysis. They can serve as the foundation for more complex operations in data analysis.