gpt4 book ai didi

go - 按行值对 [][]string(二维 slice )进行分组

转载 作者:数据小太阳 更新时间:2023-10-29 03:08:12 28 4
gpt4 key购买 nike

我正在使用 go 中的二维字符串 slice ,我想按“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] 元素的输入字符串 slice ,然后将其转换回二维 slice 。这将完成您的工作:

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]]
}

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

关于go - 按行值对 [][]string(二维 slice )进行分组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57447157/

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com