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)
}