Golang: convert slices into map

前端 未结 4 1509
抹茶落季
抹茶落季 2021-01-07 18:20

Is there an easy/simple means of converting a slice into a map in Golang? Like converting an array into hash in perl is easy to do with simple assignment like %hash =

4条回答
  •  囚心锁ツ
    2021-01-07 19:21

    There is currently no way to do it the perl way. You just have to iterate the slice, and place the slice elements in your map, e.g. as the map's key:

    func main() {
        var elements []string
        var elementMap map[string]string
        elements = []string{"abc", "def", "fgi", "adi"}
    
        // initialize map
        elementMap = make(map[string]string)
    
        // put slice values into map
        for _, s := range elements {  
            elementMap[s] = s
            // or just keys, without values: elementMap[s] = ""
        }
    
        // print map
        for k := range elementMap {
            fmt.Println(k)
        }
    }
    

    Depending on what you want to do, you have to keep one thing in mind: map keys are unique, so if your slice contains duplicate strings you might want to keep count by using a map[string]int:

    func main() {
        var elements []string
        var elementMap map[string]int
        elements = []string{"abc", "def", "fgi", "adi", "fgi", "adi"}
    
        // initialize map
        elementMap = make(map[string]int)
    
        // increment map's value for every key from slice
        for _, s := range elements {  
            elementMap[s]++
        }
    
        // print map
        for k, v := range elementMap {
            fmt.Println(k, v)
        }
    }
    

    And you can always wrap that functionality in a func:

    func sliceToStrMap(elements []string) map[string]string {
        elementMap := make(map[string]string)
        for _, s := range elements {
            elementMap[s] = s
        }
        return elementMap
    }
    
    func sliceToIntMap(elements []string) map[string]int {
        elementMap := make(map[string]int)
        for _, s := range elements {
            elementMap[s]++
        }
        return elementMap
    }
    

提交回复
热议问题