boxmoe_header_banner_img

Hello! 欢迎来到悠悠畅享网!

文章导读

Golang处理CSV文件怎么做 通过encoding/csv库解析结构化数据


avatar
站长 2025年8月16日 5

在 go 语言中,使用 encoding/csv 包可高效处理 csv 文件的读写操作,通过结合 os 和 bufio 包能将 csv 数据映射到结构体;读取时可用 csv.reader 逐行解析或 readall 一次性加载,推荐结合标题行建立列名索引以提升可维护性,写入时使用 csv.writer 并注意调用 flush 或 writeall 确保数据写入,同时支持自定义分隔符等配置,适用于结构化数据处理且需关注类型转换与错误处理,最终实现安全高效的 csv 操作。

Golang处理CSV文件怎么做 通过encoding/csv库解析结构化数据

在 Go 语言中,处理 CSV 文件非常方便,标准库中的

encoding/csv

包提供了完整的读写功能。结合

os

bufio

等包,可以高效地解析结构化数据并映射到 Go 的结构体中。下面详细介绍如何使用

encoding/csv

来读取和写入 CSV 文件,并将数据结构化。


一、读取 CSV 文件并解析为结构体

最常见的需求是将 CSV 文件中的每一行转换为 Go 的结构体。假设我们有一个用户信息的 CSV 文件

users.csv

,内容如下:

Name,Age,Email Alice,30,alice@example.com Bob,25,bob@example.com Charlie,35,charlie@example.com

我们可以定义一个结构体来映射这些字段:

立即学习go语言免费学习笔记(深入)”;

type User struct {     Name  string     Age   int     Email string }

然后使用

csv.Reader

读取文件并解析:

package main  import (     "encoding/csv"     "fmt"     "io"     "log"     "os"     "strconv" )  func main() {     file, err := os.Open("users.csv")     if err != nil {         log.Fatal("无法打开文件:", err)     }     defer file.Close()      reader := csv.NewReader(file)      // 跳过标题行     _, err = reader.Read()     if err != nil {         log.Fatal("读取标题行失败:", err)     }      var users []User     for {         record, err := reader.Read()         if err == io.EOF {             break         }         if err != nil {             log.Fatal("读取行失败:", err)         }          age, err := strconv.Atoi(record[1])         if err != nil {             log.Fatal("年龄转换失败:", err)         }          user := User{             Name:  record[0],             Age:   age,             Email: record[2],         }         users = append(users, user)     }      // 打印结果     for _, u := range users {         fmt.Printf("%+vn", u)     } }

注意:csv.Reader.Read() 每次返回一个 []string,需要手动转换类型(如 int、float 等)。


二、使用

ReadAll

一次性读取所有数据

如果数据量不大,可以使用

reader.ReadAll()

一次性读取所有行:

records, err := reader.ReadAll() if err != nil {     log.Fatal("读取所有数据失败:", err) }  var users []User for i, record := range records {     if i == 0 {         continue // 跳过标题     }     age, _ := strconv.Atoi(record[1])     users = append(users, User{         Name:  record[0],         Age:   age,         Email: record[2],     }) }

这种方式更简洁,但不适合大文件,会占用较多内存。


三、按列名自动映射结构体(使用

Read

+ 标题行)

更灵活的方式是读取标题行,然后根据列名索引映射字段,避免硬编码索引:

header, err := reader.Read() if err != nil {     log.Fatal("读取标题失败:", err) }  // 建立列名到索引的映射 headerMap := make(map[string]int) for i, name := range header {     headerMap[name] = i }  for {     record, err := reader.Read()     if err == io.EOF {         break     }     if err != nil {         log.Fatal("读取行失败:", err)     }      age, _ := strconv.Atoi(record[headerMap["Age"]])      user := User{         Name:  record[headerMap["Name"]],         Age:   age,         Email: record[headerMap["Email"]],     }     users = append(users, user) }

这种方式更健壮,即使 CSV 列顺序变化也能正确解析。


四、写入结构化数据到 CSV 文件

使用

csv.Writer

可以将结构体数据写回 CSV 文件:

file, err := os.Create("output.csv") if err != nil {     log.Fatal("创建文件失败:", err) } defer file.Close()  writer := csv.NewWriter(file) defer writer.Flush()  // 写入标题 writer.Write([]string{"Name", "Age", "Email"})  // 写入数据 for _, u := range users {     writer.Write([]string{u.Name, strconv.Itoa(u.Age), u.Email}) }

也可以批量写入:

var records [][]string records = append(records, []string{"Name", "Age", "Email"}) for _, u := range users {     records = append(records, []string{u.Name, strconv.Itoa(u.Age), u.Email}) } writer.WriteAll(records)

五、自定义分隔符和其他选项

默认情况下,

csv.Reader

使用逗号

,

作为分隔符。如果 CSV 使用其他分隔符(如分号

;

),可以自定义:

reader := csv.NewReader(file) reader.Comma = ';' // 设置分隔符为分号

其他可配置项包括:

  • reader.Comment

    :设置注释字符(如

    #

  • reader.FieldsPerRecord

    :验证每行字段数量(-1 表示不验证)

  • reader.TrimLeadingSpace

    :是否忽略字段前的空格


小结

  • 使用
    encoding/csv

    可以高效处理 CSV 文件

  • 读取时注意类型转换(尤其是数字)
  • 建议使用标题行映射字段,提高代码可维护性
  • 大文件建议逐行读取,避免内存溢出
  • 写入时注意使用
    writer.Flush()

    WriteAll

基本上就这些常用操作,不复杂但容易忽略细节,比如错误处理和类型转换。熟练掌握后,处理结构化 CSV 数据会非常高效。



评论(已关闭)

评论已关闭