Go 读取 CSV 文件|极客教程

Go 读取 CSV 文件

简介

CSV（逗号分隔值）是一种常用的文件格式，广泛用于存储和传输表格数据。CSV 文件由纯文本组成，通过逗号或其他分隔符来分隔不同字段的值。在 Go 语言中，我们可以使用标准库 encoding/csv 来读取和处理 CSV 文件。

在本文中，我们将详细介绍如何使用 Go 语言读取 CSV 文件的步骤和技巧，并提供一些示例代码来说明。

步骤

1. 导入库

首先，我们需要导入 Go 语言的 encoding/csv 包。在代码的开头添加以下代码行：

import (
    "encoding/csv"
    "os"
)

2. 打开 CSV 文件

接下来，我们需要打开要读取的 CSV 文件。使用 os 包中的 Open 函数来打开文件，并确保检查错误。以下是打开文件的示例代码：

file, err := os.Open("data.csv")
if err != nil {
    panic(err)
}
defer file.Close()

这段代码将打开名为 “data.csv” 的 CSV 文件，并使用 defer 关键字延迟关闭文件。

3. 创建 CSV Reader

有了打开的文件，我们需要创建一个 csv.Reader 对象来读取文件中的数据。以下是创建 csv.Reader 对象的示例代码：

reader := csv.NewReader(file)

4. 读取数据行

现在，我们可以使用 csv.Reader 对象的 Read 方法来逐行读取数据。Read 方法返回一个 []string 切片，其中每个元素对应一行的字段值。以下是一个读取数据行的示例代码：

for {
    record, err := reader.Read()
    if err == io.EOF {
        break
    }
    if err != nil {
        panic(err)
    }

    // 处理行数据
    fmt.Println(record)
}

此示例代码会将每一行的字段值打印出来。Read 方法每次调用都会返回下一行的字段值，直到到达文件末尾（io.EOF）。在读取完所有行之后，我们使用 break 语句退出循环。

5. 处理字段值

每一行的字段值都被存储在一个 []string 切片中，我们可以根据需要对其进行处理。根据 CSV 文件的格式，可以使用切片的索引访问每个字段的值。

以下是一个示例代码，将每行的第一列作为字符串打印出来：

for {
    record, err := reader.Read()
    if err == io.EOF {
        break
    }
    if err != nil {
        panic(err)
    }

    // 处理行数据
    fmt.Println(record[0])
}

完整示例代码

package main

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

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

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

    // 读取数据行
    for {
        record, err := reader.Read()
        if err == io.EOF {
            break
        }
        if err != nil {
            panic(err)
        }

        // 处理行数据
        fmt.Println(record)
    }
}