【golang】mapのvalueでsort

key, valueを保持する構造体を作って、sort interfaceを満たすように Len(), Swap(), Less()メソッドを実装してあげる。 これはstackoverflowに載っていた。

// A data structure to hold key/value pairs                                                              
type Pair struct {                                                                                       
    Key   string                                                                                         
    Value int                                                                                         
}                                                                                                        
                                                                                                         
// A slice of pairs that implements sort.Interface to sort by values                                     
type PairList []Pair                                                                                     
                                                                                                         
func (p PairList) Len() int           { return len(p) }                                                  
func (p PairList) Swap(i, j int)      { p[i], p[j] = p[j], p[i] }                                        
func (p PairList) Less(i, j int) bool { return p[i].Value < p[j].Value }

で、PairListの各要素にPairを詰めてあげれば、sortできると。

func rankByWordCount(wordFrequencies map[string]int) PairList{
  pl := make(PairList, len(wordFrequencies))
  i := 0
  for k, v := range wordFrequencies {
    pl[i] = Pair{k, v}
    i++
  }
  sort.Sort(sort.Reverse(pl))
  return pl
}