将函数应用于golang列表中所有元素的快捷方式

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

这是一个老问题，但在我的 Google 搜索中是热门问题，我发现我认为对 OP 和其他任何到达这里并寻找相同内容的人都有帮助的信息。

有一个更短的方法，虽然你必须自己编写地图函数。

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

请参阅此 Go by Example 页面底部附近的 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 2

我已经创建了一个带有 Map 和 Filter 方法的小型实用程序包，因为泛型已在 1.18 中引入 :)

https://pkg.go.dev/github.com/sa-/slicefunk

示例用法

package main

import (
    "fmt"

    sf "github.com/sa-/slicefunk"
)

func main() {
    original := []int{1, 2, 3, 4, 5}
    newArray := sf.Map(original, func(item int) int { return item + 1 })
    newArray = sf.Map(newArray, func(item int) int { return item * 3 })
    newArray = sf.Filter(newArray, func(item int) bool { return item%2 == 0 })
    fmt.Println(newArray)
}

Answer 3

我会按照你的做法做，只是稍微调整一下以修复错别字

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 4

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

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})