Go 读取CSV文件|极客教程

Go 读取CSV文件

CSV（Comma-Separated Values）是一种常见的电子表格文件格式，以逗号作为字段分隔符。在Go语言中，可以通过使用内置的 “encoding/csv” 包来读取CSV文件。本文将详细介绍如何在Go中读取CSV文件并进行数据处理。

读取CSV文件

在Go中读取CSV文件非常简单，我们可以使用 “encoding/csv” 包提供的 Reader 类型来读取并解析CSV文件。首先，我们需要使用 “os” 包打开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)

    // 读取CSV文件中的数据
    records, err := reader.ReadAll()
    if err != nil {
        fmt.Println("Error reading CSV:", err)
        return
    }

    // 打印CSV文件中的数据
    for _, row := range records {
        fmt.Println(row)
    }
}

在上面的示例代码中，我们首先打开名为 “data.csv” 的CSV文件，并创建一个CSV Reader对象来读取文件内容。然后使用 ReadAll() 方法来获取整个CSV文件的数据，并将其存储在一个二维切片中。最后，我们遍历二维切片并打印出每一行的数据。

在运行上述代码之前，需要先创建一个名为 “data.csv” 的CSV文件，并确保该文件包含一些数据，例如：

name,age,email
Alice,30,alice@example.com
Bob,25,bob@example.com

运行上述代码后，输出应该如下:

[name age email]
[Alice 30 alice@example.com]
[Bob 25 bob@example.com]

自定义CSV解析器

除了使用CSV Reader外，我们还可以自定义CSV解析器来处理特定的CSV格式。例如，如果CSV文件使用分号 ‘;’ 作为字段分隔符，我们可以自定义一个函数来解析该格式的CSV文件。

package main

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

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

    // 创建一个自定义的CSV Reader
    reader := csv.NewReader(file)
    reader.Comma = ';'

    // 读取CSV文件中的数据
    records, err := reader.ReadAll()
    if err != nil {
        fmt.Println("Error reading CSV:", err)
        return
    }

    // 打印CSV文件中的数据
    for _, row := range records {
        fmt.Println(row)
    }
}

在上面的示例代码中，我们首先打开名为 “data_custom.csv” 的CSV文件，并创建一个CSV Reader对象。然后通过设定 reader.Comma 属性为 ‘;’ 来指定字段分隔符为分号。接着使用 ReadAll() 方法来获取整个CSV文件的数据，将其存储在一个二维切片中，并最后打印出每一行的数据。

在运行上述代码之前，需要先创建一个名为 “data_custom.csv” 的CSV文件，并确保该文件使用分号 ‘;’ 作为字段分隔符，例如：

name;age;email
Alice;30;alice@example.com
Bob;25;bob@example.com

运行上述代码后，输出应该如下:

[name age email]
[Alice 30 alice@example.com]
[Bob 25 bob@example.com]

结论

通过使用 “encoding/csv” 包中提供的功能，我们可以轻松地读取和解析CSV文件，并对其中的数据进行处理。在处理CSV文件时，我们可以根据实际需要自定义解析器，以适应不同的CSV格式。

Go 读取CSV文件