[C++] 평균, 표준편차 구하기

평균 및 표준편차 계산 함수

#include <vector>

// 평균 계산 함수
template<class T>
double Mean(std::vector<T> array)
{
    double sum = 0.0;

    for (int i = 0; i < array.size(); i++)
        sum += array[i];

    return sum / array.size();
}

// 표준 편차 계산 함수
template<class T>
double StandardDeviation(std::vector<T> array)
{
    double meanValue = Mean<T>(array);

    // 배열 요소가 1개밖에 없을 때는
    // NaN(숫자가 아님)이라는 의미로
    // sqrt(-1.0) 을 반환
    int size = array.size();
    if (size < 2)
    {
        return sqrt(-1.0);
    }

    double sum = 0.0;

    for (int i = 0; i < size; i++) {
        double diff = array[i] - meanValue;
        sum += diff * diff;
    }

    return sqrt(sum / (size - 1));
}

테스트

int main()
{
    std::vector<int> array;
    array.push_back(5);
    array.push_back(2);
    array.push_back(7);
    array.push_back(8);
    array.push_back(4);
    array.push_back(1);
    array.push_back(10);
    array.push_back(9);

    double meanValue = Mean<int>(array);
    double stddevValue = StandardDeviation<int>(array);

    std::cout << "meanValue : " << meanValue << std::endl;
    std::cout << "stddevValue : " << stddevValue << std::endl;
}

결과

meanValue : 5.75
stddevValue : 3.28416