如何将 gota 数据框写入 csv？

对于一个Golang开发者来说，牢固扎实的基础是十分重要的，golang学习网就来带大家一点点的掌握基础知识点。今天本篇文章带大家了解《如何将 gota 数据框写入 csv？》，主要介绍了，希望对大家的知识积累有所帮助，快点收藏起来吧，否则需要时就找不到了！

我发现了很多通过传入 [][] 字符串写入 csv 的代码示例。 （如下所示）：

package main

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

var data = [][]string{
    {"row 1", "30"}, 
    {"row 2", "60"},
    {"row 3", "90"}}

func main() {
    file, err := os.create("tutorials_technology.csv")
    if err != nil {
        log.fatal(err)
    }
    defer file.close()

    w := csv.newwriter(file) 

    for _, value := range data {
        if err := w.write(value); err != nil {
            log.fatalln("error writing record to csv: ", err)
        }
    }
    w.flush()
}

但是，我还没有找到任何代码示例来说明如何使用 gota dataframe.writecsv() 函数写入 csv。在 gota dataframe 文档中，没有写入 csv 的示例，但还有一个读取 csv 的示例。

数据帧函数 writecsv() 需要 io.writer{} 接口的输入。我不知道如何满足这一点。

以下内容无效

writer := csv.newwriter(f)

df.writecsv(writer) // todo this writer needs to be a []byte writer

我已经为此工作了很长一段时间。有人有任何线索吗？

我已经考虑过将我的 gota 数据帧转换为 [][]string 类型，但这有点不方便，因为我使用包的 loadstructs() 函数将我的数据放入 gota 数据帧中，并且我已经在 a 中读取了一些 csv在将它们放入结构之前以半自定义方式。

所以我可以编写一个函数来将我的结构转换为 [][]string 格式，但我觉得这非常乏味，而且我确信必须有更好的方法。事实上，我确信存在，因为数据帧类型具有 writecsv() 方法，但我只是不知道如何使用它。

这是我的结构

type CsvLine struct {
    Index  int
    Date   string
    Symbol string
    Open   float64
    High   float64
    Low    float64
    Close  float64
    // Volume float64
    // Market_Cap float64
}

type File struct {
    Rows []CsvLine
}

免责声明：我是一个 golang 新手。我只使用 go 几个月，这是我第一次尝试写入文件。我没有与 io.writer 接口进行太多交互，但我听说它非常有用。

是的，我经常查看 golang.org 博客，我读过“effective go”并且不断引用它。

解决方案

原来我误解了 io.writer 接口，并且不明白 os.create() 函数返回什么。

事实证明，代码比我想象的更简单、更容易。

这是工作代码示例：

df := dataframe.LoadStructs(file.Rows)

f, err := os.Create(outFileName)
if err != nil {
    log.Fatal(err)
}

df.WriteCSV(f)

以上就是《如何将 gota 数据框写入 csv？》的详细内容，更多关于的资料请关注golang学习网公众号！