2024年最新Golang中读写CSV文件的全面指南_golang csv(2)，字节跳动技术总监自爆

qq_38197819

于 2024-05-11 15:39:02 发布

阅读量587

点赞数 11

分类专栏：程序员文章标签： go 学习面试

本文链接：https://blog.csdn.net/qq_38197819/article/details/138718239

版权

程序员专栏收录该内容

56 篇文章 0 订阅

订阅专栏

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上Go语言开发知识点，真正体系化！

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

如果你需要这些资料，可以戳这里获取

for {
    record, err := reader.Read()
    if err == io.EOF {
        break
    }
    if err != nil {
        fmt.Println("读取记录失败:", err)
        return
    }
    // 处理记录
}

在上述代码中，我们使用一个无限循环来持续读取CSV记录，直到遇到文件结束（EOF）为止。每次调用Read()方法，它会返回一个字符串切片，其中每个元素都是CSV记录的一个字段。最后，我们可以在循环中对记录进行处理，如打印、解析等。

写入CSV记录：如果我们想将数据写入CSV文件，则需要创建一个CSV Writer对象，并使用其Write()方法写入记录。

file, err := os.Create("output.csv")
if err != nil {
    fmt.Println("创建文件失败:", err)
    return
}
defer file.Close()

writer := csv.NewWriter(file)
defer writer.Flush()

record := []string{"Alice", "21", "F"}
err = writer.Write(record)
if err != nil {
    fmt.Println("写入记录失败:", err)
    return
}

在上述代码中，我们使用os.Create()函数创建一个名为output.csv的文件，并将其传递给csv.NewWriter()函数创建一个CSV Writer对象。类似于读取CSV文件时，我们需要在写入结束时关闭文件和刷新缓冲区，以确保数据被正确写入。

二、使用第三方库处理CSV文件

除了标准库之外，还有许多第三方库可以在Golang中处理CSV文件。这些库通常提供更多功能和灵活性，使CSV文件的处理更加高效和方便。下面介绍两个受欢迎的第三方库：

1. Gocarina/gocsv

Gocarina/gocsv是一个功能强大的CSV处理库，提供了读取、写入和转换CSV文件的能力。它具有以下特点：

支持高级功能：包括自定义类型映射、标记解析、选择性解析等。
具有更好的性能：相对于标准库，Gocarina/gocsv提供了更好的性能和效率。
简单易用：提供了一组简洁明了的API，使CSV文件的处理变得简单易用。

要使用Gocarina/gocsv库，您需要按照以下步骤进行设置：

安装Gocarina/gocsv库：在终端中执行以下命令安装Gocarina/gocsv库。

go get github.com/Gocarina/gocsv

导入所需的包：在代码中导入github.com/Gocarina/gocsv和其他所需的包。

import (
    "github.com/Gocarina/gocsv"
    "os"
)

创建结构体：使用结构体定义CSV文件中的记录结构。

type Person struct {
    Name   string `csv:"name"`
    Age    int    `csv:"age"`
    Gender string `csv:"gender"`
}

在上面的示例中，我们定义了一个名为Person的结构体，其中的字段使用csv标签指定了他们在CSV文件中对应的列名。
4. 读取CSV文件：使用gocsv.UnmarshalFile函数从CSV文件中读取记录。

file, err := os.OpenFile("data.csv", os.O_RDWR|os.O_CREATE, os.ModePerm)
if err != nil {
    fmt.Println("打开文件失败:", err)
    return
}
defer file.Close()

var people []\*Person
if err := gocsv.UnmarshalFile(file, &people); err != nil {
    fmt.Println("读取文件失败:", err)
    return
}

在上述代码中，我们使用os.OpenFile函数打开一个名为data.csv的文件，并将其传递给gocsv.UnmarshalFile函数来读取CSV记录。通过传递一个指向记录切片的指针，我们可以将文件中的记录存储在people变量中。
5. 写入CSV文件：使用gocsv.MarshalFile函数将记录写入CSV文件。

file, err := os.OpenFile("output.csv", os.O_RDWR|os.O_CREATE, os.ModePerm)
if err != nil {
    fmt.Println("创建文件失败:", err)
    return
}
defer file.Close()

people := []\*Person{
    {Name: "Alice", Age: 21, Gender: "F"},
    {Name: "Bob", Age: 25, Gender: "M"},
}
if err := gocsv.MarshalFile(&people, file); err != nil {
    fmt.Println("写入文件失败:", err)
    return
}

在上述代码中，我们使用os.OpenFile函数创建一个名为output.csv的文件，并将其传递给gocsv.MarshalFile函数来写入CSV记录。通过传递一个记录切片的指针，我们可以将记录写入文件中。

Gocarina/gocsv还提供了其他一些方便的功能，例如选择性解析、自定义类型映射、标记解析等，使CSV文件的处理更加灵活和便捷。

2. GoCSV

GoCSV是另一个流行的CSV处理库，提供了丰富的功能和灵活性。它具有以下特点：

支持读取和写入CSV文件：GoCSV提供了直观且易于使用的API来读取和写入CSV文件。
提供行级别的操作：可以对每行数据进行操作，例如过滤、更新、删除等。
支持自定义解析器和写入器：允许您创建自定义的解析器和写入器，以符合特定的需求。

要使用GoCSV库，您需要按照以下步骤进行设置：

安装GoCSV库：在终端中执行以下命令安装GoCSV库。

go get github.com/gocarina/gocsv

导入所需的包：在代码中导入github.com/gocarina/gocsv和其他所需的包。

import (
    "github.com/gocarina/gocsv"
    "os"
)

创建结构体：使用结构体定义CSV文件中的记录结构。

type Person struct {
    Name   string `csv:"name"`
    Age    int    `csv:"age"`
    Gender string `csv:"gender"`
}

file, err := os.OpenFile("data.csv", os.O_RDWR|os.O_CREATE, os.ModePerm)
if err != nil {
    fmt.Println("打开文件失败:", err)
    return
}
defer file.Close()

var people []\*Person
if err := gocsv.UnmarshalFile(file, &people); err != nil {
    fmt.Println("读取文件失败:", err)
    return
}

file, err := os.OpenFile("output.csv", os.O_RDWR|os.O_CREATE, os.ModePerm)
if err != nil {
    fmt.Println("创建文件失败:", err)
    return
}
defer file.Close()

people := []\*Person{
    {Name: "Alice", Age: 21, Gender: "F"},
    {Name: "Bob", Age: 25, Gender: "M"},
}
if err := gocsv.MarshalFile(&people, file); err != nil {
    fmt.Println("写入文件失败:", err)
    return
}

GoCSV还提供了其他一些方便的功能，如行级操作、自定义解析器和写入器等，使CSV文件的处理更加灵活和强大。

案例

下面我将给您提供 3 个使用 Gocarina/gocsv 库和 GoCSV 库处理 CSV 文件的案例。

案例1：读取和打印 CSV 文件中的记录

下面是一个使用 Gocarina/gocsv 库读取 data.csv 文件，并打印出每一条记录的示例：

package main

import (
    "encoding/csv"
    "fmt"
    "os"

    "github.com/Gocarina/gocsv"
)

type Person struct {
    Name   string `csv:"name"`
    Age    int    `csv:"age"`
    Gender string `csv:"gender"`
}

func main() {
    file, err := os.OpenFile("data.csv", os.O_RDWR|os.O_CREATE, os.ModePerm)
    if err != nil {
        fmt.Println("打开文件失败:", err)
        return


![img](https://img-blog.csdnimg.cn/img_convert/7b810c429529d9b2beb6cc6da00ae56e.png)
![img](https://img-blog.csdnimg.cn/img_convert/9bc2a625eeae4ba33fe1608fcb4bbb57.png)
![img](https://img-blog.csdnimg.cn/img_convert/7d33cb1377821daac1fe7b96d0f2b6b2.png)

**既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上Go语言开发知识点，真正体系化！**

**由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新**

**[如果你需要这些资料，可以戳这里获取](https://bbs.csdn.net/topics/618658159)**

存中...(img-4gRodh0A-1715413119699)]
[外链图片转存中...(img-gyhkylkE-1715413119699)]

**既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上Go语言开发知识点，真正体系化！**

**由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新**

**[如果你需要这些资料，可以戳这里获取](https://bbs.csdn.net/topics/618658159)**

qq_38197819

关注

11
点赞
踩
8

收藏

觉得还不错? 一键收藏
0
评论
2024年最新Golang中读写CSV文件的全面指南_golang csv(2)，字节跳动技术总监自爆

在上述代码中，我们使用一个无限循环来持续读取CSV记录，直到遇到文件结束（EOF）为止。每次调用Read()方法，它会返回一个字符串切片，其中每个元素都是CSV记录的一个字段。最后，我们可以在循环中对记录进行处理，如打印、解析等。Write()在上述代码中，我们使用函数创建一个名为output.csv的文件，并将其传递给函数创建一个CSV Writer对象。类似于读取CSV文件时，我们需要在写入结束时关闭文件和刷新缓冲区，以确保数据被正确写入。
复制链接

扫一扫