我正在 go 中處理字符串的 2D 切片，我想按“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]]
1 回答
素胚勾勒不出你
TA貢獻1827條經驗 獲得超9個贊
以下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]]
}
對上面代碼的分析留作練習。:)
- 1 回答
- 0 關注
- 131 瀏覽
添加回答
舉報
0/150
提交
取消