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