最佳方式：根据map的长度，新建一个数组，遍历map逐个压入
方法1（效率很高）：
|
1
2
3
4
5
6
7
8
9
10
|
func getKeys1(m map[int]int) []int { // 数组默认长度为map长度,后面append时,不需要重新申请内存和拷贝,效率很高 j := 0 keys := make([]int, len(m)) for k := range m { keys[j] = k j++ } return keys} |
方法2（效率很高）：
|
1
2
3
4
5
6
7
8
|
func getKeys2(m map[int]int) []int { // 数组默认长度为map长度,后面append时,不需要重新申请内存和拷贝,效率很高 keys := make([]int, 0, len(m)) for k := range m { keys = append(keys, k) } return keys} |
其他方式：
方法3（效率较低）：
|
1
2
3
4
5
6
7
8
|
func getKeys3(m map[int]int) []int { // 注意：由于数组默认长度为0，后面append时，需要重新申请内存和拷贝，所以效率较低 keys := []int{} for k := range m { keys = append(keys, k) } return keys} |
方法4（效率极低）：
|
1
2
3
4
5
|
func getKeys4(m map[int]int) int { // 注意:虽然此写法简洁,但MapKeys函数内部操作复杂,效率极低 keys := reflect.ValueOf(m).MapKeys() return len(keys)} |
实验结果如图（可以看到方法1和方法2效率最高，内存操作也最少）：
完整代码如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
package test import ( "reflect" "testing") // 初始化mapfunc initMap() map[int]int { m := map[int]int{} for i := 0; i < 10000; i++ { m[i] = i } return m} func getKeys1(m map[int]int) []int { // 数组默认长度为map长度,后面append时,不需要重新申请内存和拷贝,效率较高 j := 0 keys := make([]int, len(m)) for k := range m { keys[j] = k j++ } return keys} func getKeys2(m map[int]int) []int { // 数组默认长度为map长度,后面append时,不需要重新申请内存和拷贝,效率较高 keys := make([]int, 0, len(m)) for k := range m { keys = append(keys, k) } return keys} // 初始化默认func getKeys3(m map[int]int) []int { // 注意：由于数组默认长度为0，后面append时，需要重新申请内存和拷贝，所以效率较低 keys := []int{} for k := range m { keys = append(keys, k) } return keys} // 使用反射func getKeys4(m map[int]int) int { // 注意:虽然此写法简洁,但MapKeys函数内部操作复杂,效率极低 keys := reflect.ValueOf(m).MapKeys() return len(keys)} func BenchmarkMapkeys1(b *testing.B) { // 初始化map m := initMap() b.ResetTimer() for i := 0; i < b.N; i++ { getKeys1(m) }}func BenchmarkMapkeys2(b *testing.B) { // 初始化map m := initMap() b.ResetTimer() for i := 0; i < b.N; i++ { getKeys2(m) }} func BenchmarkMapkeys3(b *testing.B) { // 初始化map m := initMap() b.ResetTimer() for i := 0; i < b.N; i++ { getKeys3(m) }} func BenchmarkMapkeys4(b *testing.B) { // 初始化map m := initMap() b.ResetTimer() for i := 0; i < b.N; i++ { getKeys4(m) }} |
补充：Golang踩坑——判断map中是否有key
最近在实习，下班回去十点多了，再加上比较懒，快有两个月没写东西了。
今天在开发一个模块的时候是接着上一个人的写的，好不容易各种配置写好了开始跑，发现他踩了一个很容易踩的坑。
把bug抽出来单独写了个文件
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
package main import ( "log") type agent struct { id int str string} var m map[int]*agent func main() { m = make(map[int]*agent) a := &agent{ id: 1, str: "hello", } log.Println(a) var ok bool if a, ok = m[1]; ok { log.Println("ok") } m[1] = a log.Println(a) log.Println(a.str)} |
运行一下
可以看到报了无效指针的问题（invalid memory address or nil pointer dereference）。
从上面的两个log可以看到，最开始指针是有值的，第二个却没了，原因就出现在那个if判断那里。
if判断是想通过查看map知道agent是否已经存在，这个地方的错误是a,ok = map; 我们想通过ok判断是否存在，但是在这里还有个a，这会导致有一个对a赋值的操作，也就是说如果map里有这个值，那没事，如果没有的话a就被赋值成了nil了。
在下面打印的时候也就成了nil，取它的属性的时候会告诉我们找了个空指针的属性，这肯定找不到。
我们把a换成_，再来看结果：
ok了。
以上为个人经验，希望能给大家一个参考，也希望大家多多支持服务器之家。如有错误或未考虑完全的地方，望不吝赐教。
原文链接：https://blog.csdn.net/yzf279533105/article/details/94010954
