对字符串进行行值分组

怎么入门Golang编程？需要学习哪些知识点？这是新手们刚接触编程时常见的问题；下面golang学习网就来给大家整理分享一些知识点，希望能够给初学者一些帮助。本篇文章就来介绍《对字符串进行行值分组》，涉及到，有需要的可以收藏一下

我正在使用 go 中的二维字符串切片，我想按“a”列值对它们进行分组，但我无法弄清楚。

我尝试使用 gota 数据框，但它也没有像 pandas 中可用的分组依据。

    input := [][]string{
        []string{"b", "3", "2.9", "5.3"},
        []string{"a", "4", "5.1", "9.1"},
        []string{"b", "4", "6.0", "5.3"},
        []string{"c", "3", "6.0", "5.5"},
        []string{"a", "2", "7.1", "9.2"},
    }

我想要这样的输出。

[[b 3 2.9 5.3 4 6.0 5.3] [a 4 5.1 9.1 2 7.1 9.2] [c 3 6.0 5.5]]

解决方案

以下 group() 函数利用映射来收集具有相同 [0] 元素的输入字符串切片，然后将其转换回 2d 切片。这将完成您的工作：

func group(input [][]string) (output [][]string) {
    tmp := map[string][]string{}
    for _, slice := range input {
        if len(slice) <= 1 {
            continue
        }
        tmp[slice[0]] = append(tmp[slice[0]], slice[1:]...)
    }
    for k := range tmp {
        v := append([]string{k}, tmp[k]...)
        output = append(output, v)
    }
    return
}

func main() {
    input := [][]string{
        []string{"b", "3", "2.9", "5.3"},
        []string{"a", "4", "5.1", "9.1"},
        []string{"b", "4", "6.0", "5.3"},
        []string{"c", "3", "6.0", "5.5"},
        []string{"a", "2", "7.1", "9.2"},
    }
    fmt.Println(group(input)) // [[a 4 5.1 9.1 2 7.1 9.2] [c 3 6.0 5.5] [b 3 2.9 5.3 4 6.0 5.3]]
}

上面代码的分析留作练习。 :)

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。