主页 / 问题 / 1596229
Accepted
Андрей Алексеев
Asked:2024-10-10 13:23:50 +0000 UTC2024-10-10 13:23:50 +0000 UTC

如何按键数组中指定的顺序对地图进行排序？

  • 772

是的map，或者更确切地说map[string]int

默认情况下，据我了解，Golang它按字母顺序按键对元素进行排序。如何map按给定顺序按键排序？

这段代码：

package main

import (
    "fmt"
)

func main() {
    output := make(map[string]int)

    basket := map[string]int{
        "orange": 5, 
        "apple": 7,
        "mango": 3,
        "strawberry": 9,
    }

    //нужный порядок
    keys := []string{
        "strawberry",
        "mango",
        "orange",
    }

    for _, k := range keys {
        output[k] = basket[k]
    }

    fmt.Printf("вывод output : %+v \n", output)
}

将生成带有按字母顺序排列的键的行：

mango, orange, strawberry
golang

2 个回答

  1. 对于打印，map标准库确实按键排序。这就是fmtsort.Sort函数的作用。

    但是Go中键/值的枚举是故意随机的。每次迭代映射中的键时，都可以获得一个新序列：https://go.dev/play/p/wgHyoYTGC-t

    package main

import "fmt"

func main() {
    basket := map[string]int{
        "orange":     5,
        "apple":      7,
        "mango":      3,
        "strawberry": 9,
        "mellon":     1,
        "pineapple":  4,
        "papaya":     1,
    }

    for i := 0; i < 5; i++ {
        fmt.Println("Попытка №", i)
        for k := range basket {
            fmt.Printf("  ключ: %+v \n", k)
        }
    }
}

    Попытка № 0
  ключ: apple
  ключ: mango
  ключ: strawberry
  ключ: mellon
  ключ: pineapple
  ключ: papaya
  ключ: orange
Попытка № 1
  ключ: orange
  ключ: apple
  ключ: mango
  ключ: strawberry
  ключ: mellon
  ключ: pineapple
  ключ: papaya
Попытка № 2
  ключ: strawberry
  ключ: mellon
  ключ: pineapple
  ключ: papaya
  ключ: orange
  ключ: apple
  ключ: mango
.
.
.

    要对地图进行排序，您需要创建一个键值对数组并对其进行排序：https://go.dev/play/p/8SbeIbCTleJ

    package main

import (
    "fmt"
    "slices"
    "strings"
)

type Pair[K any, V any] struct {
    Key   K
    Value V
}

type PairList[K any, V any] []Pair[K, V]

type PairComparator[K any, V any] func(Pair[K, V], Pair[K, V]) int

func SortMap[K comparable, V any](m map[K]V, cf PairComparator[K, V]) PairList[K, V] {
    result := make(PairList[K, V], len(m))
    i := 0
    for k, v := range m {
        result[i] = Pair[K, V]{k, v}
        i += 1
    }
    slices.SortFunc(result, func(a Pair[K, V], b Pair[K, V]) int { return cf(a, b) })
    return result
}

func SortMapKeys[K comparable, V any](m map[K]V, cf PairComparator[K, V]) []K {
    result := make([]K, len(m))
    i := 0
    for k := range m {
        result[i] = k
        i += 1
    }
    slices.SortFunc(result, func(a K, b K) int { return cf(Pair[K, V]{a, m[a]}, Pair[K, V]{b, m[b]}) })
    return result
}

func main() {
    basket := map[string]int{
        "orange":     5,
        "apple":      7,
        "mango":      3,
        "strawberry": 9,
        "mellon":     1,
        "pineapple":  4,
        "papaya":     1,
    }

    sortedBasket := SortMap(basket, basketSorter)
    for _, pair := range sortedBasket {
        fmt.Println("sorted basket: ", pair.Key, pair.Value)
    }
}

func basketSorter(p1, p2 Pair[string, int]) int {
    if p1.Value < p2.Value {
        return -1
    }
    if p1.Value > p2.Value {
        return 1
    }
    return strings.Compare(p1.Key, p2.Key)
}

    结果：

    sorted basket:  mellon 1
sorted basket:  papaya 1
sorted basket:  mango 3
sorted basket:  pineapple 4
sorted basket:  orange 5
sorted basket:  apple 7
sorted basket:  strawberry 9
    • 3
  2. Best Answer

    map- 这是键：值形式的未排序集合，不保留插入顺序，使用排序键完成搜索（但这不准确）。常规字典在 Python（直到某个版本，似乎是 3.6）和其他编程语言中的行为方式大致相同。

    要保留插入顺序，您需要编写自己的类，或者使用保留插入顺序的现成包：orderedmap。

    • 1

相关问题