Golang中Map大小排序

时间:2022-05-28 IT范儿人气:3

Golang中的 map 默认是无序的。

起因

最近项目中有这样一个需求：

根据用户当前的坐标点，获取该用户附近的预设城市名称。

这里有一个注意点是，假设这些支持的城市名称是预设的，所以就不能直接通过地图类api根据坐标点获取所在城市名称了。

想到的解决思路是：

获取这几个预设城市的坐标点
App端获取用户当前坐标点
分别计算得到该用户坐标点距离各个预设城市的坐标点距离
然后计算得到其中距离最小的一项
这个坐标点对应的城市就是所求

探索

经过前期计算，在上面的第 3 步操作后我得到了下面的结果：

result := map[string]float64{
"城市A": 2334.20,
"城市B": 1992.33,
"城市C": 500.26,
"城市D": 10.39,
"城市E": 333.33,
}

我们知道，Golang中 Map 是无序的。所以当我们使用 for-range 循环时：

for k, v := range result {  
   fmt.Printf("key: %v value: %v \n", k, v)  
}

结果可能是：

// 第一种可能结果：
key: 城市B value: 1992.33
key: 城市C value: 500.26
key: 城市D value: 10.39
key: 城市E value: 333.33
key: 城市A value: 2334.2

// 第二种可能结果：
key: 城市E value: 333.33
key: 城市A value: 2334.2
key: 城市B value: 1992.33
key: 城市C value: 500.26
key: 城市D value: 10.39

// 第三种可能结果：
key: 城市E value: 333.33
key: 城市A value: 2334.2
key: 城市B value: 1992.33
key: 城市C value: 500.26
key: 城市D value: 10.39

所以，我们不能按照 key 或者 value 来进行排序。

实现

但Golang中切片 Slice 是有序的。我们可以结果使用 Slice 来实现对 Map 的排序。

第一步

我们先将上面的 map 转换成一个 slice :

type KVPair struct {  
   Key string  
 Val float64  
}  
  
tmpList := make([]KVPair, 0)  
  
for k, v := range result {  
   tmpList = append(tmpList, KVPair{Key: k, Val: v})  
}

上面创建了一个结构体切片，然后将 map 的值添加到切片中。

第二步

在 go1.8 之后，引入了 sort.Slice() 方法，可以实现对 slice 进行排序，我们只需要传入一个比较函数即可：

sort.Slice(tmpList, func(i, j int) bool {  
   return tmpList[i].Val < tmpList[j].Val // 升序  
})

第三步

然后，我们对排序后的 slice 进行 for-range 遍历：

for _, pair := range tmpList {  
   fmt.Printf("key: %v value: %v \n", pair.Key, pair.Val)  
}

// 结果：
key: 城市D value: 10.39 
key: 城市E value: 333.33 
key: 城市C value: 500.26 
key: 城市B value: 1992.33 
key: 城市A value: 2334.2

可以看到，排序后的 slice 第一项就是我们想要的结果。

如果我们想要获取其中 value 值最大的一项，只需要更改 sort.Slice 中的比较方法接口：

sort.Slice(tmpList, func(i, j int) bool {  
   return tmpList[i].Val > tmpList[j].Val // 降序  
 //return tmpList[i].Val < tmpList[j].Val // 升序})

总结

上面测试的完整代码如下：

package main  
  
import (  
   "fmt"  
 "sort")  
  
var result = map[string]float64{  
   "城市A": 2334.20,  
 "城市B": 1992.33,  
 "城市C": 500.26,  
 "城市D": 10.39,  
 "城市E": 333.33,  
}

func main() {

   type KVPair struct {  
      Key string  
      Val float64  
   }  
  
   tmpList := make([]KVPair, 0)  
  
   for k, v := range result {  
      tmpList = append(tmpList, KVPair{Key: k, Val: v})  
   }  
  
   sort.Slice(tmpList, func(i, j int) bool {  
      //return tmpList[i].Val > tmpList[j].Val // 降序  
      return tmpList[i].Val < tmpList[j].Val // 升序 
   })  
  
   for _, pair := range tmpList {  
      fmt.Printf("key: %v value: %v \n", pair.Key, pair.Val)  
   }  
}

总结

加载全部内容