在 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中，这个方法叫做functional options.

这是 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)
}

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

在 gota 中设置列​​名

Setting column names in gota

go

dataframe

gota

在 gota 中设置列名