将函数应用于 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

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 2

使用 go1.18+，您可以编写更清晰的通用 Map 函数：

func Map[T, V any](ts []T, fn func(T) V) []V {
    result := make([]V, len(ts))
    for i, t := range ts {
        result[i] = fn(t)
    }
    return result
}

用法，例如：

input := []int{4, 5, 3}
outputInts := Map(input, func(item int) int { return item + 1 })
outputStrings := Map(input, func(item int) string { return fmt.Sprintf("Item:%d", item) })

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

这是一个古老的问题，但在我的 Google 搜索中排名靠前，我找到了我认为对 OP 和其他到达这里寻找相同内容的人有帮助的信息。

有一种更短的方法，尽管您必须自己编写 map 函数。

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

查看Map此 Go by Example 页面底部附近的功能：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 5

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

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