dataframe - 在 gota 中设置列​​名

Question

我有一个没有定义列名的数据集。如何使用 gota 以编程方式分配这些？相比之下，Pandas有df.rename.

假设我的数据如下所示：

1,2,3
4,5,6
7,8,9

我基本上想要这个在我的数据框中：

A,B,C
1,2,3
4,5,6
7,8,9

Answer 1

您可以调用ReadCSV加载选项。在 go 中，这种方法称为功能选项。

这是 gota 中的示例：

func main() {
    f, err := os.Open("sample2.csv")
    if err != nil {
        panic(err.Error())
    }
    defer f.Close()

    names := dataframe.Names("A", "B", "C")
    noHeader := dataframe.HasHeader(false)

    df := dataframe.ReadCSV(f, names, noHeader)
    r := df.Records()

    fmt.Println(r)
}

函数Load*和Read*gota 都有LoadOption：
https ://pkg.go.dev/github.com/go-gota/gota/dataframe