从地图中获取密钥切片

Question

在Go中，有没有更简单/更好的方法从地图中获取一部分密钥？

目前，我正在遍历地图并将关键点复制到一个切片：

i := 0
keys := make([]int, len(mymap))
for k := range mymap {
    keys[i] = k
    i++
}

Answer 1

您也可以通过包“reflect”中struct Value的方法< code>MapKeys获取类型为< code>[]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]
}

Answer 2

举个例子，

package main

func main() {
    mymap := make(map[int]string)
    keys := make([]int, 0, len(mymap))
    for k := range mymap {
        keys = append(keys, k)
    }
}

为了提高Go的效率，最小化内存分配很重要。

Answer 3

这是一个老问题，但这是我的两分钱。PeterSO的回答略为简洁，但效率略低。你已经知道它会有多大，所以你甚至不需要使用append：

keys := make([]int, len(mymap))

i := 0
for k := range mymap {
    keys[i] = k
    i++
}

在大多数情况下，这可能不会有太大的区别，但这并不会增加太多工作，在我的测试中(使用一个包含1，000，000个随机< code>int64键的映射，然后用每种方法生成十次键数组)，直接分配数组成员比使用append快20%左右。

尽管设置容量消除了重新分配，但追加仍然需要做额外的工作来检查您是否已达到每个追加的容量。