GolangCSV包使用教程：读写CSV文件方法

想知道如何在 Golang 中高效读写 CSV 文件吗？本文将深入探讨 `encoding/csv` 包的强大功能，助你轻松玩转 CSV 数据的导入导出。无论是标准格式还是自定义分隔符，`encoding/csv` 都能完美支持。我们将详细讲解如何使用 `csv.NewReader` 从 `io.Reader` 读取数据，并通过 `ReadAll()` 一次性读取或 `Read()` 逐行读取，灵活应对不同大小的文件。同时，还将介绍 `csv.NewWriter` 的使用方法，教你如何向 `io.Writer` 写入数据，并使用 `Write()` 或 `WriteAll()` 写入数据，最后通过 `Flush()` 确保数据安全落盘。此外，本文还将分享如何通过设置 `Comma` 字段修改分隔符，以及处理读取时可能遇到的引号与特殊字符，助你避坑，编写更健壮的 CSV 处理代码。

Go语言中encoding/csv包用于读写CSV文件，支持标准格式及自定义分隔符。1. 使用csv.NewReader从io.Reader读取数据，可调用ReadAll()一次性读取或Read()逐行读取；2. 使用csv.NewWriter向io.Writer写入数据，通过Write()或WriteAll()写入并需调用Flush()确保数据落盘；3. 可设置Comma字段修改分隔符，如分号常用于欧洲地区；4. 读取时自动处理引号与特殊字符，错误需检查error返回值，尤其注意ParseError和Flush必要性。

在Go语言中，encoding/csv 包提供了读取和写入CSV（逗号分隔值）文件的简单方式。它能处理标准格式的CSV数据，也支持自定义分隔符。下面介绍如何使用该包进行常见的读写操作。

读取CSV文件

使用 csv.NewReader 可以从任何实现了 io.Reader 接口的对象中读取CSV数据，比如文件或字符串。

基本步骤如下：

• 打开文件并创建 csv.Reader
• 调用 ReadAll() 一次性读取所有行，或使用 Read() 逐行读取
• 处理返回的二维字符串切片

示例代码：

package main

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

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

    reader := csv.NewReader(file)
    records, err := reader.ReadAll()
    if err != nil {
        panic(err)
    }

    for _, record := range records {
        fmt.Println(record) // 每一行是一个[]string
    }
}

如果想逐行读取以节省内存（适合大文件），可以这样：

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

写入CSV文件

使用 csv.NewWriter 可将数据写入任何实现了 io.Writer 的目标，如文件或缓冲区。

步骤包括：

• 创建 csv.Writer
• 使用 Write() 写入单行，或 WriteAll() 写入多行
• 调用 w.Flush() 确保数据写入底层设备

示例代码：

package main

import (
    "encoding/csv"
    "os"
)

func main() {
    file, err := os.Create("output.csv")
    if err != nil {
        panic(err)
    }
    defer file.Close()

    writer := csv.NewWriter(file)
    defer writer.Flush()

    records := [][]string{
        {"Name", "Age", "City"},
        {"Alice", "25", "Beijing"},
        {"Bob", "30", "Shanghai"},
    }

    err = writer.WriteAll(records)
    if err != nil {
        panic(err)
    }
}

处理自定义分隔符

默认情况下，csv.Reader 和 csv.Writer 使用逗号作为字段分隔符。如果需要使用其他字符（如分号或制表符），可以手动设置。

例如，使用分号作为分隔符：

reader := csv.NewReader(file)
reader.Comma = ';'

writer := csv.NewWriter(file)
writer.Comma = ';'

常见场景包括欧洲国家导出的CSV文件常使用分号，注意根据实际数据调整。

错误处理与注意事项

CSV操作可能遇到多种错误，比如格式错误、引号不匹配等。建议始终检查返回的 error 值。

• 读取时若某行引号未闭合，会返回 csv.ParseError
• 写入的数据不需要手动加引号，包会自动处理包含逗号或换行的内容
• 写完数据后务必调用 Flush()，否则可能丢失最后几行

基本上就这些。encoding/csv 使用简单，配合 os.Open 和 os.Create 能轻松完成文件读写。对结构化数据导入导出来说非常实用。

今天关于《GolangCSV包使用教程：读写CSV文件方法》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于csv文件,io.Reader,encoding/csv,csv.NewReader,io.Writer的内容请关注golang学习网公众号！