Golang基础教程——map使用篇

本文始发于个人公众号：TechFlow，原创不易，求个关注

今天是golang专题的第7篇文章，我们来聊聊golang当中map的用法。

map这个数据结构我们经常使用，存储的是key-value的键值对。在C++/java当中叫做map，在Python中叫做dict。这些数据结构的名称虽然不经相同，背后的技术支撑也不一定一样，比如说C++的map是红黑树实现的，Java中的hashmap则是通过hash表。但是使用起来的方法都差不多，除了Java是通过get方法获取键值之外，C++、Python和golang都是通过方括号获取的。

声明与初始化

golang中的map声明非常简单，我们用map关键字表示声明一个map，然后在方括号内填上key的类型，方括号外填上value的类型。

var m map[string] int

这样我们就声明好了一个map。

但是要注意，这样声明得到的是一个空的map，map的零值是nil，可以理解成空指针。所以我们不能直接去操作这个m，否则会得到一个panic。

panic: assignment to entry in nil map

panic在golang当中表示非常严重不可恢复的错误，可以恢复的错误有些类似于Java或者是其他语言当中的异常，当异常出现的时候，我们可以选择handle住它们，让程序不崩溃继续运行。而那些非常严重，无法handle的异常在golang当中称为panic。

golang当中的异常处理机制和其他语言相差很大，整体的逻辑和内核都不太一样。当然这个是一个比较大的话题，我们这里可以简单将它理解成error就行了。

回到map上来，我们声明了一个map之后，想要使用它还需要对它进行初始化。使用它的方法也很简单，就是使用make方法创建出一个实例来。它的用法和之前通过make创建元组非常类似：

m = make(map[string] int)

// 我们还可以指定创建出来的map的存储能力的大小

m = make(map[string] int, 100)

我们也可以在声明的时候把初始化也写上：

var m = map[string] int {"abc": 3, "ccd": 4}

当然也可以通过赋值运算符，直接make出一个空的map来：

m := make(map[string] int)

增删改查

map创建好了当然是要用的，整体使用起来和Python当中的dict比较像，比较简单直观，没有太多弯弯绕的东西。我们一个一个来看，首先是map的添加元素。map的添加元素直接用方括号赋值即可：

m["abc"] = 4

同样，我们需要保证这里的m经过初始化，否则也会包nil的panic。如果key值在map当中已经存在，那么会自动替换掉原本的key。也就是说map的更新和添加元素都是一样的，都是通过这种方式。如果不存在就是添加，否则则是更新。

删除元素也很简单，和Python当中类似，通过delete关键字删除。

delete(m, "abc")

当我们删除key的时候，如果是其他的语言，我们需要判断这个key值是否存在，否则的话不能删除，或者是会引起异常。在golang当中并不会，对这点做了优化。如果要删除的key值原本就不在map当中，那么当我们调用了delete之后，什么也不会发生。但是有一点，必须要保证传入的map不为nil，否则也会引起panic。

最后，我们看下元素的查找。对于Java和Python来说我们都是通过一些判断语句来进行判断的，比如java的话是containsKey，Python的话用in操作符。在golang当中我们则是直接通过方括号进行查询，那么这就有了一个问题，如果key不在其中怎么办？

如果是其他语言，我们直接访问一个不存在的key是会抛出异常的，但是在golang当中不会触发panic，因为它会额外返回一个bool类型的元素表示元素是否查找到。所以我们可以同时用两个变量去接收，如果第二个变量为True的话，就说明查找成功了。

进一步，我们还可以将这个逻辑和if的初始化操作合在一起：

if val, ok := m["1234"]; ok {

    fmt.Println(val)

}

这里的ok就表示查找是否成功，这也是golang当中map查找的惯用写法。

最后，我们看一个实际运用map的例子，通过map来生成统计字符串当中单词数量的wordCount：

package main

import (

 "golang.org/x/tour/wc"

 "strings"

)

func WordCount(s string) map[string]int {

 cnt := make(map[string]int)

    // 通过Split方法拆分字符串

 for _, str:= range strings.Split(s){

        // 直接++即可，golang会自动填充

  cnt[str]++

 }

 return cnt

}

func main() {

 wc.Test(WordCount)

}

总结

到这里，关于golang当中map的使用就算是介绍完了。我们可以发现，map一如既往地体现了golang语法精简的特点。比如通过返回error的操作省略了判断元素是否存在map当中的操作，刚开始的时候会觉得有些不太适应，但是接触多了之后，会发现这些都是有套路的。golang的套路就是精简，能省就省，能简单绝不复杂。

这一点不仅在map上体现，在其他特性上也是一样。在后续的内容当中，我们还会继续感知这一点。

如果喜欢本文，可以的话，请点个关注，给我一点鼓励，也方便获取更多文章。