go 读取csv文件|极客笔记

go 读取csv文件

在Go语言中，可以使用encoding/csv包来读取和写入CSV文件。CSV（Comma-Separated Values）是一种常见的文件格式，通常用于存储表格数据。在本文中，我们将学习如何读取CSV文件并对其进行处理。

读取CSV文件

要读取CSV文件，首先需要使用os包打开文件，然后使用encoding/csv包中的NewReader函数创建一个CSV Reader对象。接下来，我们可以使用Read方法逐行读取文件中的数据。

以下是一个简单的示例，演示如何读取名为data.csv的CSV文件，并输出每行的数据：

package main

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

func main() {
    // 打开CSV文件
    file, err := os.Open("data.csv")
    if err != nil {
        fmt.Println("Error opening file:", err)
        return
    }
    defer file.Close()

    // 创建CSV Reader对象
    reader := csv.NewReader(file)

    // 逐行读取数据
    records, err := reader.ReadAll()
    if err != nil {
        fmt.Println("Error reading file:", err)
        return
    }

    // 输出数据
    for _, record := range records {
        for _, value := range record {
            fmt.Printf("%s\t", value)
        }
        fmt.Println()
    }
}

假设data.csv文件内容如下：

Name,Age,City
Alice,25,New York
Bob,30,Los Angeles
Charlie,35,Chicago

运行上面的代码，将得到如下输出：

Name    Age     City    
Alice   25      New York    
Bob     30      Los Angeles
Charlie 35      Chicago

自定义CSV Reader

除了使用Read和ReadAll方法读取整个文件之外，我们还可以自定义CSV Reader来逐个字段读取数据。以下是一个自定义CSV Reader的示例：

package main

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

func main() {
    // 打开CSV文件
    file, err := os.Open("data.csv")
    if err != nil {
        fmt.Println("Error opening file:", err)
        return
    }
    defer file.Close()

    // 创建CSV Reader对象
    reader := csv.NewReader(file)

    // 逐个字段读取数据
    for {
        record, err := reader.Read()
        if err != nil {
            fmt.Println("Error reading file:", err)
            return
        }

        if record == nil {
            break
        }

        for _, value := range record {
            fmt.Printf("%s\t", value)
        }
        fmt.Println()
    }
}

运行上面的代码，将得到与前面相同的输出。

处理CSV数据

一旦我们读取了CSV文件的数据，就可以对其进行各种处理。例如，我们可以根据条件过滤数据，对数据进行排序，或者进行统计分析。

以下是一个简单的示例，演示如何计算data.csv文件中Age列的平均值：

package main

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

func main() {
    // 打开CSV文件
    file, err := os.Open("data.csv")
    if err != nil {
        fmt.Println("Error opening file:", err)
        return
    }
    defer file.Close()

    // 创建CSV Reader对象
    reader := csv.NewReader(file)

    // 逐个字段读取数据
    sum := 0
    count := 0
    for {
        record, err := reader.Read()
        if err != nil {
            fmt.Println("Error reading file:", err)
            return
        }

        if record == nil {
            break
        }

        age, err := strconv.Atoi(record[1])
        if err != nil {
            continue
        }

        sum += age
        count++
    }

    if count > 0 {
        average := float64(sum) / float64(count)
        fmt.Printf("Average Age: %.2f\n", average)
    }
}

运行上面的代码，将得到如下输出：