Question

假设我想将一个函数应用于列表中的每个元素，然后将结果值放在另一个列表中，以便我可以立即使用它们。在python中，我会做这样的事情：

list = [1,2,3]
str = ', '.join(multiply(x, 2) for x in list)

在Go中，我做了类似的事情：

list := []int{1,2,3}
list2 := []int

for _,x := range list {
    list2 := append(list2, multiply(x, 2))
}

str := strings.Join(list2, ", ")

是否可以用更短的方式做到这一点？

Answer 1

我会像你一样做，通过一些调整来解决拼写错误

import (
    "fmt"
    "strconv"
    "strings"
)

func main() {
    list := []int{1,2,3}

    var list2 []string
    for _, x := range list {
        list2 = append(list2, strconv.Itoa(x * 2))  // note the = instead of :=
    }

    str := strings.Join(list2, ", ")
    fmt.Println(str)
}

Answer 2

这是一个老问题，但在我的谷歌搜索中是最受欢迎的，我发现我认为有助于OP和其他任何人来到这里寻找同样事情的信息。

虽然你必须自己编写地图功能，但有一种更短的方法。

在go中，func是一种类型，它允许您编写一个函数，该函数接受主题切片和函数作为输入，并在该切片上迭代，应用该函数。

请参阅此示例页面底部附近的Map函数：https://gobyexample.com/collection-functions

我已将其纳入此处以供参考：

func Map(vs []string, f func(string) string) []string {
    vsm := make([]string, len(vs))
    for i, v := range vs {
        vsm[i] = f(v)
    }
    return vsm
}

然后你这样称呼它：

fmt.Println(Map(strs, strings.ToUpper))

所以，是的：你寻找的方式越短越好，尽管它没有内置于语言本身。

Answer 3

找到一种定义通用地图数组函数的方法

func Map(t interface{}, f func(interface{}) interface{} ) []interface{} {
    switch reflect.TypeOf(t).Kind() {
    case reflect.Slice:
        s := reflect.ValueOf(t)
        arr := make([]interface{}, s.Len())
        for i := 0; i < s.Len(); i++ {
            arr[i] = f(s.Index(i).Interface())
        }
        return arr
    }
    return nil
}

origin := []int{4,5,3}
newArray := Map(origin, func(item interface{}) interface{} { return item.(int) + 1})

将函数应用于golang

3 个答案: