Golang读写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 能轻松完成文件读写。对结构化数据导入导出来说非常实用。