Go 编程 | 连载 13 - 复杂数据类型 Map

807 阅读3分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第9天,点击查看活动详情

一、Map 的定义

Go 中的 Map 是一组无序的 K-V 类型的数据,与 Python 中的字典 Dict 和 Java 中的 HashMap 结构类似。

// Map 的结构
map[string]string

Map 的定义

func main() {

   // 使用 Map 结构 函数定义 Map
   info := map[string]string{
      "name":    "Stark",
      "address": "NewYork",
   }

   fmt.Printf("%T, %v\n", info, info)

   // 使用 make 函数定义 Map
   polit := make(map[string]string)
   polit["nickname"] = "Maverick"
   polit["aircraft"] = "F18"
   fmt.Printf("%T, %v\n", polit, polit)

   // 定义空 Map
   maverick := map[string]string{}
   fmt.Printf("%T, %v", maverick, maverick)
}

执行上述代码,输出结果如下:

map[string]string, map[address:NewYork name:Stark]
map[string]string, map[aircraft:F18 nickname:Maverick]
map[string]string, map[]

Map 中的 Key

Map 中的 Key 必须是唯一的,且 Key 必须支持 ==!= 比较操作符。Map 中 Key 可以是 int、rune、指针和结构体等类型(支持 ==!=

image.png

切片不支持比较(切片只能和 nil 比较),不能作为 Map 的 Key,数组支持比较操作,可以作为 Key。

func main() {

   structKey01 := structKey{name: "structKey01"}
   fmt.Printf("%T", structKey01)
   info := map[structKey]string{
      structKey01: "Stark",
   }
   fmt.Printf("%T, %v\n", info, info)
   
   info1 := map[interfaceKey]string{
      interfaceKey: "Stark",
   }
   fmt.Printf("%T, %v\n", info1, info1)
}

type structKey struct {
   name string
}

type interfaceKey interface {
   
}

接口和结构体可作为 Key,前提是所包含的属性必须都是可以进行 == 或者 != 比较的,如果包含一个属性不可以进行 == 或者 != 比较,便不能作为 Key。

二、Map 的操作

  • 增加或者修改键值对
  • 查询,如果查询的键不存在返回 空字符串
func main() {

   // 使用 Map 结构 函数定义 Map
   info := map[string]string{
      "name":    "Stark",
      "address": "NewYork",
   }

   fmt.Printf("%p, %T, %v\n", info, info, info)

   // 增加 KV
   info["nickname"] = "IRONMAN"
   // 修改
   info["name"] = "Tony Stark"
   fmt.Printf("%p, %T, %v\n", info, info, info)

   // 查询,Key 存在
   fmt.Println("名字属性值是:", info["name"])
   // 查询,Key 不存在
   fmt.Println("技能属性值是:", info["weapon"])

   // 添加一个空值的 KV
   info["balance"] = ""
   fmt.Println("余额属性值是:", info["balance"])

}

执行上述代码,输出结果如下:

0xc00010e180, map[string]string, map[address:NewYork name:Stark]
0xc00010e180, map[string]string, map[address:NewYork name:Tony Stark nickname:IRONMAN]
名字属性值是: Tony Stark
技能属性值是: 
余额属性值是: 

根据上述代码,查询时如果 Key 不存在返回空字符串,如果 Key 的值是空字符串,那么查询时返回的也是空字符串,显然通过返回字符串是否为空来判断 Key 是否存在是不准确的。

应该如何判断 Key 是否存在?

Map 通过 [Key] 进行取值时会有两个返回值,第一个值是 Key 对应的 Value,第二个值是布尔值,如果为 true 表示 Key 存在,否则表示 Key 不存在。

func main() {
    
   // 其余代码保持不变 
   weaponVal, ok1 := info["weapon"]
   fmt.Printf("weapon 属性值是:%v, 是否存在:%v\n", weaponVal, ok1)
   balanceVal, ok2 := info["balance"]
   fmt.Printf("balance 属性值是:%v, 是否存在:%v\n", balanceVal, ok2)

}

执行上述代码,输出结果如下:

weapon 属性值是:, 是否存在:false
balance 属性值是:, 是否存在:true

删除 Key 需要使用到内置的 delete 函数,在执行删除操作时,如果 Key 不存在也不会有异常

func main() {

   // 其余代码保持不变 
   delete(info, "nickname")
   fmt.Printf("%p, %T, %v\n", info, info, info)
   delete(info, "weapon")
   fmt.Printf("%p, %T, %v\n", info, info, info)

}

执行上述代码,输出结果如下:

0xc000098180, map[string]string, map[address:NewYork balance: name:Tony Stark]
0xc000098180, map[string]string, map[address:NewYork balance: name:Tony Stark]

遍历 Map 可以使用 for ... range

func main() {

   // 使用 Map 结构 函数定义 Map
   info := map[string]string{
      "name":    "Stark",
      "address": "NewYork",
   }

   for k, v := range info {
      fmt.Println(k, ":", v)
   }

}

执行上述代码,输出结果如下:

name : Tony Stark
address : NewYork

Map 的遍历是无序的