如何用Golang处理CSV文件读写_Golang CSV文件操作实践-Golang-PHP中文网

如何用Golang处理CSV文件读写_Golang CSV文件操作实践

P粉602998670

发布： 2025-11-24 21:39:06

原创

724人浏览过

Golang通过encoding/csv包可高效读写CSV文件。使用csv.NewReader读取文件内容，结合os.Open和bufio.Reader提升性能；支持逐行读取大文件以节省内存。写入时用csv.NewWriter并将数据写入文件，需调用Flush确保数据落盘。可通过手动转换实现结构体与CSV记录映射，或使用第三方库增强功能。示例涵盖表头处理、数据遍历及错误管理，适用于常见CSV操作场景。

如何用golang处理csv文件读写_golang csv文件操作实践

处理CSV文件在数据导入导出、日志分析和批量操作中非常常见。Golang标准库encoding/csv提供了简洁高效的接口来读写CSV文件，结合os和bufio包可以轻松完成文件操作。下面通过实际示例说明如何用Golang进行CSV的读取与写入。

读取CSV文件

使用csv.NewReader可以从文件或任意io.Reader读取CSV数据。通常配合os.Open打开文件，并用bufio.Reader提升效率。

以下是一个读取本地CSV文件的示例：

package main

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

func readCSV(filename string) {
    file, err := os.Open(filename)
    if err != nil {
        log.Fatal("无法打开文件:", err)
    }
    defer file.Close()

    reader := csv.NewReader(file)
    // 读取所有记录
    records, err := reader.ReadAll()
    if err != nil {
        log.Fatal("读取CSV失败:", err)
    }

    for i, record := range records {
        fmt.Printf("第%d行: %v\n", i+1, record)
    }
}

func main() {
    readCSV("data.csv")
}

登录后复制

如果CSV有表头，可以跳过第一行或单独解析字段名：

立即学习“go语言免费学习笔记（深入）”；

if len(records) > 0 {
    header := records[0]  // 第一行是表头
    data := records[1:]   // 剩余是数据
}

登录后复制

逐行读取大文件

对于大CSV文件，一次性读取可能占用过多内存。建议使用Read()方法逐行处理：

轻幕

轻幕是一个综合性短视频制作平台，诗词、故事、小说等一键成片转视频，让内容传播更生动！

查看详情

func readCSVLineByLine(filename string) {
    file, err := os.Open(filename)
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close()

    reader := csv.NewReader(file)
    for {
        record, err := reader.Read()
        if err == io.EOF {
            break
        }
        if err != nil {
            log.Fatal("读取行失败:", err)
        }
        fmt.Println(record)
    }
}

登录后复制

写入CSV文件

使用csv.NewWriter可以将数据写入文件。写完后记得调用w.Flush()确保数据落盘。

func writeCSV(filename string) {
    file, err := os.Create(filename)
    if err != nil {
        log.Fatal("创建文件失败:", err)
    }
    defer file.Close()

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

    // 写入表头
    header := []string{"姓名", "年龄", "城市"}
    if err := writer.Write(header); err != nil {
        log.Fatal("写入表头失败:", err)
    }

    // 写入多行数据
    data := [][]string{
        {"张三", "25", "北京"},
        {"李四", "30", "上海"},
        {"王五", "28", "广州"},
    }
    if err := writer.WriteAll(data); err != nil {
        log.Fatal("写入数据失败:", err)
    }
}

登录后复制

若需逐行写入，可用writer.Write([]string{...})代替WriteAll。

结构体与CSV映射

虽然标准库不直接支持结构体标签，但可通过辅助函数实现转换。例如：

type User struct {
    Name string
    Age  int
    City string
}

func writeUsersToCSV(users []User, filename string) {
    file, err := os.Create(filename)
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close()

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

    // 手动转换结构体为字符串切片
    for _, u := range users {
        record := []string{u.Name, fmt.Sprintf("%d", u.Age), u.City}
        if err := writer.Write(record); err != nil {
            log.Fatal("写入记录失败:", err)
        }
    }
}

登录后复制

也可以使用第三方库如gocarina/gocsv实现自动绑定。

基本上就这些。Golang处理CSV简单直接，适合大多数场景。注意错误处理和资源释放，避免文件句柄泄漏。对于复杂需求可引入外部库增强功能。

以上就是如何用Golang处理CSV文件读写_Golang CSV文件操作实践的详细内容，更多请关注php中文网其它相关文章！