Mendapatkan sepotong kunci dari peta

Apakah ada cara yang lebih sederhana / lebih baik untuk mendapatkan sepotong kunci dari peta di Go?

Saat ini saya mengulangi peta dan menyalin kunci ke sebuah irisan:

i := 0
keys := make([]int, len(mymap))
for k := range mymap {
    keys[i] = k
    i++
}

Sebagai contoh,

package main

func main() {
    mymap := make(map[int]string)
    keys := make([]int, 0, len(mymap))
    for k := range mymap {
        keys = append(keys, k)
    }
}

Agar efisien di Go, penting untuk meminimalkan alokasi memori.

Ini pertanyaan lama, tapi ini dua sen saya. Jawaban PeterSO sedikit lebih ringkas, tetapi sedikit kurang efisien. Anda sudah tahu seberapa besar itu sehingga Anda bahkan tidak perlu menggunakan append:

keys := make([]int, len(mymap))

i := 0
for k := range mymap {
    keys[i] = k
    i++
}

Dalam kebanyakan situasi mungkin tidak akan membuat banyak perbedaan, tapi itu tidak banyak bekerja, dan dalam pengujian saya (menggunakan peta dengan 1.000.000 int64kunci acak dan kemudian menghasilkan array kunci sepuluh kali dengan masing-masing metode), itu tentang 20% lebih cepat untuk menetapkan anggota array secara langsung daripada menggunakan append.

Meskipun pengaturan kapasitas menghilangkan realokasi, append masih harus melakukan pekerjaan ekstra untuk memeriksa apakah Anda telah mencapai kapasitas pada setiap append.

Anda juga dapat mengambil larik kunci dengan jenis []Valuemenurut metode MapKeysstruct Valuedari paket "mencerminkan":

package main

import (
    "fmt"
    "reflect"
)

func main() {
    abc := map[string]int{
        "a": 1,
        "b": 2,
        "c": 3,
    }

    keys := reflect.ValueOf(abc).MapKeys()

    fmt.Println(keys) // [a b c]
}

Cara yang lebih baik untuk melakukan ini adalah dengan menggunakan append:

keys = []int{}
for k := range mymap {
    keys = append(keys, k)
}

Selain itu, Anda kurang beruntung — Go bukan bahasa yang sangat ekspresif.

Saya membuat patokan samar pada tiga metode yang dijelaskan dalam tanggapan lain.

Jelas pra-alokasi potongan sebelum menarik kunci lebih cepat daripada appending, tetapi mengejutkan, reflect.ValueOf(m).MapKeys()metode ini jauh lebih lambat daripada yang terakhir:

❯ go run scratch.go
populating
filling 100000000 slots
done in 56.630774791s
running prealloc
took: 9.989049786s
running append
took: 18.948676741s
running reflect
took: 25.50070649s

Berikut kodenya: https://play.golang.org/p/Z8O6a2jyfTH (menjalankannya di taman bermain batal dengan mengklaim bahwa terlalu lama, jadi, jalankan secara lokal.)

Kunjungi https://play.golang.org/p/dx6PTtuBXQW

package main

import (
    "fmt"
    "sort"
)

func main() {
    mapEg := map[string]string{"c":"a","a":"c","b":"b"}
    keys := make([]string, 0, len(mapEg))
    for k := range mapEg {
        keys = append(keys, k)
    }
    sort.Strings(keys)
    fmt.Println(keys)
}