在Go中是否有更简单/更好的方法从地图中获取键片?
目前我在地图上迭代并复制键到一个切片:
i := 0
keys := make([]int, len(mymap))
for k := range mymap {
keys[i] = k
i++
}
当前回答
一个更好的方法是使用append:
keys = []int{}
for k := range mymap {
keys = append(keys, k)
}
除此之外，你就不走运了——围棋不是一种很有表现力的语言。
其他回答
Vinay Pai的答案的通用版本(1.18+)。
// MapKeysToSlice extract keys of map as slice,
func MapKeysToSlice[K comparable, V any](m map[K]V) []K {
keys := make([]K, len(m))
i := 0
for k := range m {
keys[i] = k
i++
}
return keys
}
访问https://play.golang.org/p/dx6PTtuBXQW
package main
import (
"fmt"
"sort"
)
func main() {
mapEg := map[string]string{"c":"a","a":"c","b":"b"}
keys := make([]string, 0, len(mapEg))
for k := range mapEg {
keys = append(keys, k)
}
sort.Strings(keys)
fmt.Println(keys)
}
我对其他回答中描述的三种方法做了一个粗略的基准。
显然，在提取键之前预分配切片比追加快，但令人惊讶的是，reflect.ValueOf(m).MapKeys()方法明显比后者慢:
❯ go run scratch.go
populating
filling 100000000 slots
done in 56.630774791s
running prealloc
took: 9.989049786s
running append
took: 18.948676741s
running reflect
took: 25.50070649s
代码如下:https://play.golang.org/p/Z8O6a2jyfTH (在操场上运行它会中止，因为它花费的时间太长，所以，在本地运行它。)
有一个很酷的lib叫lo
一个lodash风格的Go库，基于Go 1.18+泛型(映射，过滤器，包含，查找…)
有了这个库，你可以做很多方便的操作，如映射，过滤，减少等等。还有一些地图类型的助手
Keys
创建一个映射键数组。
keys := lo.Keys[string, int](map[string]int{"foo": 1, "bar": 2})
// []string{"bar", "foo"}
值
创建映射值的数组。
values := lo.Values[string, int](map[string]int{"foo": 1, "bar": 2})
// []int{1, 2}
你也可以从"reflect"包中获取一个类型为[]Value的键数组，方法为MapKeys of struct Value:
package main
import (
"fmt"
"reflect"
)
func main() {
abc := map[string]int{
"a": 1,
"b": 2,
"c": 3,
}
keys := reflect.ValueOf(abc).MapKeys()
fmt.Println(keys) // [a b c]
}
