登录
首页 >  Golang >  Go问答

在 Go 中如何使用字符串分隔符将数据写入 CSV 文件

来源:stackoverflow

时间:2024-03-21 13:51:43 493浏览 收藏

在 Go 中使用 "encoding/csv" 包处理 CSV 输出时,csv.Writer 默认使用单个字符作为分隔符。若要使用自定义分隔符(如 "|-|"),需要编写自定义写入函数。本文提供了一个示例代码,展示了如何使用自定义分隔符写入 CSV 文件,该代码将记录字段使用指定分隔符连接后写入 CSV Writer,并实现了 Writer 接口的 Write 方法。

问题内容

我正在使用“encoding/csv”csv.Writer 来处理 csv 输出

如何使用“|-|”等字符串分隔符将数据写入 CSV 文件

csv.Writer 只能使用单个字符作为分隔符,因为 Writer.Comma 被定义为 int32, 不支持字符串分隔符


正确答案


用于使用“|-|”作为分隔符,您必须编写自己的自定义写入函数。我尝试了下面的代码并得到如下输出:

*包主

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

type customcsvwriter struct {
    writer    *csv.writer
    delimiter string
}

func newcustomcsvwriter(w *csv.writer, delimiter string) *customcsvwriter {
    return &customcsvwriter{
        writer:    w,
        delimiter: delimiter,
    }
}

func (c *customcsvwriter) write(record []string) error {
    // join the fields using the custom delimiter
    joinedrecord := strings.join(record, c.delimiter)

    // write the joined record to the csv writer
    return c.writer.write([]string{joinedrecord})
}

func main() {
    file, err := os.create("output.csv")
    if err != nil {
        fmt.println("error creating file:", err)
        return
    }
    defer file.close()

    writer := csv.newwriter(file)

    // set the custom delimiter
    customdelimiter := "|-|"
    customwriter := newcustomcsvwriter(writer, customdelimiter)

    data := [][]string{
        {"alice", "25", "new york"},
        {"bob", "30", "san francisco"},
        {"charlie", "28", "chicago"},
    }

    for _, record := range data {
        if err := customwriter.write(record); err != nil {
            fmt.println("error writing record:", err)
            return
        }
    }

    customwriter.writer.flush()

    if err := customwriter.writer.error(); err != nil {
        fmt.println("error flushing csv writer:", err)
        return
    }

    fmt.println("csv data written to output.csv")
}*

我希望这就是您所要求的。

Alice|-|25|-|New York
Bob|-|30|-|San Francisco
Charlie|-|28|-|Chicago

今天关于《在 Go 中如何使用字符串分隔符将数据写入 CSV 文件》的内容介绍就到此结束,如果有什么疑问或者建议,可以在golang学习网公众号下多多回复交流;文中若有不正之处,也希望回复留言以告知!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>