go语言中encoding/csv包用于读写csv文件,无需额外依赖。使用csv.NewReader可从文件等io.Reader源读取数据,ReadAll()适合小文件,大文件应逐行Read()避免内存溢出。csv.NewWriter写入数据时需调用Flush()确保写入磁盘。可通过设置Comma字段自定义分隔符如制表符或分号。该包自动处理含逗号、换行、引号的字段,解析和写入时会正确转义。常见操作包括跳过头部标题行:读取后忽略首条记录或单独读取header再循环剩余行。配合os、bufio使用可应对多数场景,注意错误处理与Flush调用。

在Go语言中,encoding/csv 包提供了读取和写入 CSV(逗号分隔值)文件的简单而强大的功能。它内置于标准库中,无需额外安装依赖,适合处理结构化文本数据,比如导出报表、导入配置或数据迁移等场景。下面介绍如何使用该包进行常见操作。
读取CSV文件
使用 csv.NewReader 可以从任意实现了 io.Reader 接口的数据源读取 CSV 内容,比如文件、字符串或网络响应。
示例:从文件读取所有记录:
file, err := os.Open("data.csv") if err != nil { log.Fatal(err) } defer file.Close() reader := csv.NewReader(file) records, err := reader.ReadAll() if err != nil { log.Fatal(err) } for _, record := range records { fmt.Println(record) // 每一行是一个[]String }
说明:
立即学习“go语言免费学习笔记(深入)”;
- ReadAll() 一次性读取全部行,适用于小文件。
- 对于大文件,建议使用 Read() 逐行读取,避免内存溢出:
for { record, err := reader.Read() if err == io.EOF { break } if err != nil { log.Fatal(err) } fmt.Println(record) }
写入CSV文件
使用 csv.NewWriter 将数据写入支持 io.Writer 的目标,如文件或缓冲区。
示例:写入多行数据到文件:
file, err := os.Create("output.csv") if err != nil { log.Fatal(err) } defer file.Close() writer := csv.NewWriter(file) defer writer.Flush() // 必须调用Flush,否则可能丢失数据 data := [][]string{ {"Name", "Age", "City"}, {"Alice", "25", "Beijing"}, {"Bob", "30", "Shanghai"}, } for _, record := range data { err := writer.Write(record) if err != nil { log.Fatal(err) } }
注意:Flush() 会将缓冲区中的数据真正写入底层设备,不能省略。
自定义分隔符
默认情况下,CSV 使用逗号作为字段分隔符。如果需要处理制表符或其他符号分隔的文件(如TSV),可以修改 Comma 字段。
reader := csv.NewReader(file) reader.Comma = 't' // 设置为制表符 writer := csv.NewWriter(file) writer.Comma = ';' // 使用分号分隔
确保读写双方使用相同的分隔符,否则解析会出错。
处理包含特殊字符的字段
CSV 标准允许字段中包含逗号、换行符或引号,这些内容会被自动加引号并转义。encoding/csv 能正确处理这类情况。
例如原始数据:
“John Doe”,”Engineer, Developer”,”location: “HQ””
Go 会将其解析为:
[]string{“John Doe”, “Engineer, Developer”, `Location: “HQ”`}
写入时,只要字段包含特殊字符,库会自动添加引号,无需手动处理。
跳过头部标题行
很多 CSV 文件第一行为列名。若需跳过,可在读取后忽略第一项:
records, _ := reader.ReadAll() if len(records) > 0 { header := records[0] // 保存头 data := records[1:] // 实际数据 }
或者逐行读取时单独处理首行:
header, _ := reader.Read() for { record, err := reader.Read() if err == io.EOF { break } // 处理数据行 }
基本上就这些。encoding/csv 简洁高效,配合 os.Open 和 bufio 使用,能应对大多数 CSV 场景。关键是记得检查错误、调用 Flush,以及根据实际格式设置分隔符。不复杂但容易忽略细节。