如何在Golang中使用encoding/csv处理CSV文件_Golang encoding/csv CSV文件操作方法汇总

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

如何在Golang中使用encoding/csv处理CSV文件_Golang encoding/csv CSV文件操作方法汇总

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,以及根据实际格式设置分隔符。不复杂但容易忽略细节。

上一篇
下一篇
text=ZqhQzanResources