Golang中读写CSV文件的全面指南

这篇具有很好参考价值的文章主要介绍了Golang中读写CSV文件的全面指南。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

CSV(逗号分隔值)文件是一种常见的数据存储格式,广泛应用于数据导入、导出、分析和交换等场景。在Golang中,有许多库和工具可以帮助我们读取和写入CSV文件,使数据处理变得简单而高效。本文将深入探讨如何在Golang中使用标准库以及第三方库来读写CSV文件。

一、Golang标准库的CSV处理

Golang的标准库encoding/csv包提供了一组功能强大而灵活的API,用于读取和写入CSV文件。我们可以通过下面的步骤来使用标准库处理CSV文件:

  1. 导入encoding/csv包:首先,我们需要在代码中导入encoding/csv包,通过import "encoding/csv"语句实现。

  2. 创建CSV Reader:使用标准库中的csv.NewReader()函数创建一个CSV Reader对象。我们需要传递一个io.Reader对象作为参数,该对象通常是一个文件。

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

reader := csv.NewReader(file)

在上述代码中,我们使用os.Open()函数打开一个名为data.csv的文件,并将其传递给csv.NewReader()函数创建一个CSV Reader对象。注意,我们需要在读取文件后记得关闭文件,以防止资源泄露。

  1. 读取CSV记录:通过CSV Reader对象的Read()方法来迭代读取CSV文件中的记录。每次调用Read()方法,它会返回一个记录以及可能出现的错误。
for {
    record, err := reader.Read()
    if err == io.EOF {
        break
    }
    if err != nil {
        fmt.Println("读取记录失败:", err)
        return
    }
    // 处理记录
}

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

  1. 写入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库,您需要按照以下步骤进行设置:

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

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

    import (
        "github.com/Gocarina/gocsv"
        "os"
    )
    
  3. 创建结构体:使用结构体定义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库,您需要按照以下步骤进行设置:

  1. 安装GoCSV库:在终端中执行以下命令安装GoCSV库。

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

    import (
        "github.com/gocarina/gocsv"
        "os"
    )
    
  3. 创建结构体:使用结构体定义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记录。通过传递一个记录切片的指针,我们可以将记录写入文件中。

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
    }
    defer file.Close()

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

    for _, p := range people {
        fmt.Println("Name:", p.Name)
        fmt.Println("Age:", p.Age)
        fmt.Println("Gender:", p.Gender)
        fmt.Println("--------")
    }
}

案例2:将结构体切片写入 CSV 文件

下面是一个使用 Gocarina/gocsv 库将结构体切片写入 output.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("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
    }
    fmt.Println("写入成功!")
}

案例3:使用 GoCSV 库进行行级操作

下面是一个使用 GoCSV 库进行行级操作的示例,将具有特定条件的记录写入 filtered.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
    }
    defer file.Close()

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

    filteredPeople := make([]*Person, 0)
    for _, p := range people {
        if p.Age >= 18 && p.Gender == "F" {
            filteredPeople = append(filteredPeople, p)
        }
    }

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

    if err := gocsv.MarshalFile(&filteredPeople, filteredFile); err != nil {
        fmt.Println("写入文件失败:", err)
        return
    }
    fmt.Println("写入成功!")
}

这些案例涵盖了使用 Gocarina/gocsv 库和 GoCSV 库进行 CSV 文件处理的常见场景。您可以根据自己的需求进行相应的调整和修改。希望对您有所帮助!

三、总结

CSV文件是一种常见的数据存储格式,在许多场景下被广泛使用。在Golang中,我们可以使用标准库以及一些第三方库来读取和写入CSV文件。标准库encoding/csv提供了简单而高效的API来处理CSV文件,而第三方库如Gocarina/gocsvGoCSV提供了更多的功能和灵活性,使CSV文件的处理更加便捷和强大。

无论选择使用哪个库,重要的是根据需要选择最合适的工具来处理CSV文件。这将根据项目的要求、性能需求以及个人或团队的偏好来决定。以上介绍的是Golang中常用的CSV处理方法,希望对您有所帮助。文章来源地址https://www.toymoban.com/news/detail-775843.html

到了这里,关于Golang中读写CSV文件的全面指南的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 第四章:数据操作Ⅰ 第二节:读写CSV文件

    使用read.csv函数,可以将CSV文件读入数据框,而使用write.csv()函数,则可以将数据框保存到CSV中 我们使用read.csv()的时候,其会返回一个数据框 例如: 例如:读取学分文件(以下为待读取文件)  使用head格式到R语言中取前面数据  使用tail格式到R语言中取后面数据  例如:

    2024年03月08日
    浏览(49)
  • 【头歌】——数据分析与实践-基于Python语言的文件与文件夹管理-文本 文件处理-利用csv模块进行csv文件的读写操作

    第1关 创建子文件夹 第2关 删除带有只读属性的文件 第3关 批量复制文件夹中的所有文件 未通过本题,如果您通过了本题欢迎补充到评论区,有时间我会整理进来 第1关 读取宋词文件,根据词人建立多个文件 第2关 读取宋词文件,并根据词人建立多个文件夹 第3关 读取宋词文

    2024年01月25日
    浏览(73)
  • Python 实现txt、excel、csv文件读写【附源码】

    前言 一、txt文件读写 二、excel文件读写 总结 本文介绍使用Python进行文件读写操作,包括txt文件、excel文件(xlsx、xls、csv) 编译器使用的是PyCharm read()                  # 一次性读取全部内容 readline()            # 读取第一行内容 readlines()          # 读取文本全部内

    2024年02月01日
    浏览(70)
  • 100天精通Python(数据分析篇)——第56天:Pandas读写txt和csv文件(read_csv、to_csv参数说明+代码实战)

    文本文件,主要包括csv和txt两种等,相应接口为 read_csv() 和 to_csv(),分别用于读写数

    2023年04月08日
    浏览(59)
  • 《Java面向对象程序设计》学习笔记——CSV文件的读写与处理

    ​笔记汇总: 《Java面向对象程序设计》学习笔记 笔记记录的不是非常详实,如果有补充的建议或纠错,请踊跃评论留言!!! CSV 是英文 comma-separated values 的缩写,翻译为 “逗号分隔值“。 CSV 文件可以理解为以带逗号分隔(也可以是其他简单字符分割)的纯文本形式存储

    2024年02月10日
    浏览(52)
  • Pyspark读写csv,txt,json,xlsx,xml,avro等文件

    读: 读: 写: 读: 写: 读: 写: 读: 写: 读: 写: 读: 写: 读: 写: 读: 写:

    2024年02月07日
    浏览(44)
  • python 读写 json,csv,txt,docx,xlsx,xls文件大全

    最近再处理数据的时候,总会用到读写json,csv,xlsx和xls文件代码 这里就做个总结记录一下 扩展: json.loads(line)和 json.load(line) 通用写法: 1.2.1 list类型写入 1.2.2 dict类型写入 注意:json.dump()将Python对象序列化为JSON格式,并将其写入文件中。 扩展:json.dumps()和 json.dump() 2.2.1

    2024年02月06日
    浏览(46)
  • python爬取招聘网信息并保存为csv文件

    我们以猎聘网为例 一、打开网站查找信息 进入后搜索想要爬取的岗位信息,右键选择 “检查” 进入开发者界面 点击右上角的network,选择doc 然后点击图中的搜索按钮,输入想要爬取的岗位名称,然后刷新页面,选择搜索下边的第二个 这个时候我们看到有我们需要的url,从

    2024年02月09日
    浏览(44)
  • Python(wordcloud):根据词频信息(xlsx、csv文件)绘制词云图

    本文将介绍如何用python根据词频信息(xlsx、csv文件)绘制词云图,除了绘制常规形状的词云图(比如长方形),还可以指定词云图的形状。 1、安装相关的库 2、导入相关的库 1、文件读取 本文使用的数据集是excel文件(后缀名是.xlsx),该文件包含2个字段:以及对应的

    2024年01月18日
    浏览(36)
  • 源码分享-golang的BMP文件读写库

    用于读写BMP文件的golang版源码 源码基于源码分享-golang的二进制文件读写库 https://blog.csdn.net/zhyulo/article/details/128890546 BMP文件格式可参考位图文件解析-位图(bmp)、图标(ico)与光标(cur) https://blog.csdn.net/zhyulo/article/details/85934728

    2024年01月22日
    浏览(40)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包