在dataframe-go中如何使用角括号运算符进行筛选？

一分耕耘，一分收获！既然打开了这篇文章《在dataframe-go中如何使用角括号运算符进行筛选？》，就坚持看下去吧！文中内容包含等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

我正在尝试在 dataframe-go 中做一个简单的过滤器。我改编自github示例，并尝试在过滤器函数中使用简单的>运算符，但编译器给出了invalid操作：vals [“day”] > 4（operator > not define on interface）错误（请参阅下面的我的代码）

好的，所以我将类型更改为 map[string]int64 但随后，它给出了错误： cannot 转换 func 文字 (type func(map[string]int64, int, int) (dataframe.filteraction, error))输入 dataframe.filterdataframefn。

我非常熟悉 r dataframe/table 和 pandas，但是 dataframe-go 的 api 相当复杂。我找到的唯一网络资源就是这个，但作者也表示他无法理解过滤器 api。

感谢任何帮助，谢谢！

s1 := dataframe.NewSeriesInt64("day", nil, 1, 2, 3, 4, 5, 6, 7, 8)
    s2 := dataframe.NewSeriesFloat64("sales", nil, 50.3, 23.4, 56.2, nil, nil, 84.2, 72, 89)
    df := dataframe.NewDataFrame(s1, s2)

    // Try filtering
    filterFn := dataframe.FilterDataFrameFn(
        func(vals map[string]int64, row, nRows int) (dataframe.FilterAction, error) {
        //func(vals map[interface{}]interface{}, row, nRows int) (dataframe.FilterAction, error) {
        if vals["day"] > 4 {  // <= This is where I changed things
            return dataframe.KEEP, nil
        }
        return dataframe.DROP, nil
        })

    ctx := context.Background()
    dt_filtered, _ := dataframe.Filter(ctx, df, filterFn)

    fmt.Print(dt_filtered)

解决方案

您必须传递具有所需签名的函数，因此 vals 必须是 map[interface{}]interface{}。这是必需的，因为 go 不是解释性语言，并且调用作为参数传递的函数的代码对传入的函数没有编译时知识。

由于您的值是 interface{}s，因此您必须将它们转换为正确的类型才能使用它们：

if ivalue, ok:=vals["day"].(int64); ok { // Check if the value is int64, and go ahead if so
    if ivalue > 4 { 
       return dataframe.KEEP, nil
    }
}

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《在dataframe-go中如何使用角括号运算符进行筛选？》文章吧，也可关注golang学习网公众号了解相关技术文章。