[go] CSV 파일의 데이터 평균 계산

12 Dec 2023

CSV 파일은 데이터를 저장하고 전송하는 데 매우 일반적으로 사용되는 형식입니다. 이 포스트에서는 Go 언어를 사용하여 CSV 파일에서 데이터를 읽고 평균을 계산하는 방법에 대해 알아보겠습니다.

Step 1: CSV 파일 읽기

가장 먼저, encoding/csv 패키지를 사용하여 CSV 파일을 읽습니다. 아래는 CSV 파일을 읽고 데이터를 출력하는 간단한 예제 코드입니다.

package main

import (
	"encoding/csv"
	"fmt"
	"log"
	"os"
)

func main() {
	// CSV 파일 열기
	file, err := os.Open("data.csv")
	if err != nil {
		log.Fatal(err)
	}
	defer file.Close()

	// CSV 파일 파싱
	reader := csv.NewReader(file)
	records, err := reader.ReadAll()
	if err != nil {
		log.Fatal(err)
	}

	// 데이터 출력
	fmt.Println(records)
}

Step 2: 데이터 평균 계산

이제 CSV 파일에서 읽은 데이터를 사용하여 평균을 계산해 봅시다. 아래 예제 코드는 CSV 파일에서 숫자 데이터를 읽고 이를 사용하여 평균을 계산하는 방법을 보여줍니다.

package main

import (
	"encoding/csv"
	"fmt"
	"log"
	"os"
	"strconv"
)

func main() {
	// CSV 파일 열기
	file, err := os.Open("data.csv")
	if err != nil {
		log.Fatal(err)
	}
	defer file.Close()

	// CSV 파일 파싱
	reader := csv.NewReader(file)
	records, err := reader.ReadAll()
	if err != nil {
		log.Fatal(err)
	}

	// 데이터 평균 계산
	sum := 0.0
	count := 0
	for _, row := range records {
		for _, value := range row {
			num, err := strconv.ParseFloat(value, 64)
			if err == nil {
				sum += num
				count++
			}
		}
	}

	average := sum / float64(count)
	fmt.Printf("평균: %.2f\n", average)
}

위의 예제 코드에서는 CSV 파일에서 숫자 데이터를 읽어와서 이를 합산한 뒤 평균을 계산하고 출력합니다.

이제 Go 언어를 사용하여 CSV 파일의 데이터를 읽고 평균을 계산하는 방법을 활용해보세요.

참고 문헌:

Go 언어 공식 문서