go中的python枚举

Question

假设我有斐波那契数的生成器，我想使用 enumerate(get_next_fibs(10)) 并且我想拥有对的生成器 index, number_from_generator，我正在努力寻找“命名返回值”的解决方案

这不是应该如何完成的，而是为了学习有关生成器的特定内容

package main

import "fmt"

func get_next_fibs(ii int) func() int {
    i := 0
    a, b := 0, 1
    fc := func() int {
        i++
        a, b = b, a+b
        if ii <= i {
            return -1
        }
        return a
    }
    return fc
}

func enumerate(iter func() int) func() (index, v int) {
    index := 0
    fc := func() (index, v int) {
        v := iter()
        return
        index++
    }
    return fc
}

func main() {
    iter := enumerate(get_next_fibs(10))
    // iter := get_next_fibs(10)
    fmt.Printf("iter = %T\n", iter)
    for tuple := iter(); tuple != -1; tuple = iter() {
        fmt.Println("tuple:", tuple)
    }
}

Answer 1

您在此代码示例中遇到了一些问题：

1. return 语句后不能有index++。 Use defer 如果你需要在return-ing 之后做点什么。
2. 您错过了变量阴影在 go 中的工作方式。因此，您正在尝试修改错误的 index 变量。
3. Go 没有元组。

...
func enumerate(iter func() int) func() (index, v int) {
    counter := 0
    return func() (index, v int) {
        i := counter
        counter++
        return i, iter()
    }
}
...
func main() {
    iter := enumerate(get_next_fibs(10))
    fmt.Printf("iter = %T\n", iter)
    for i, v := iter(); v != -1; i, v = iter() {
        fmt.Printf("i: %d, v: %d\n", i, v)
    }
}

Playground link