golang实现python xrange方法

前端之家收集整理的这篇文章主要介绍了golang实现python xrange方法前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

在python中,可以使用xrange方法方便地获取一个整数列表:

for@H_301_4@ v in@H_301_4@ xrange(5@H_301_4@):
    print(v)

输出

0
1
2
3
4

golang没有原生的xrange方法,不过可以使用golang的channel配合goroutine实现一个xrange方法
下面是一个简易版的:

func@H_301_4@ xRangeInt(count int@H_301_4@) chan@H_301_4@ int@H_301_4@ {
    ch := make@H_301_4@(chan@H_301_4@ int@H_301_4@)

    go@H_301_4@ func@H_301_4@() {
        for@H_301_4@ i := 0@H_301_4@; i < count; i++ {
            ch <- i
        }
        close@H_301_4@(ch)
    }()

    return@H_301_4@ ch
}

完整的xrange可以接受三个参数,xrange(start,stop,step)。比如

for@H_301_4@ v in@H_301_4@ xrange(10@H_301_4@,20@H_301_4@,2@H_301_4@):
    print(v)

输出

10
12
14
16
18

下面完善一下 golang版的 xRangeInt方法

func@H_301_4@ xRangeInt(args ...int@H_301_4@) chan@H_301_4@ int@H_301_4@ {
    if@H_301_4@ l := len@H_301_4@(args); l < 1@H_301_4@ || l > 3@H_301_4@ {
        fmt.Println("error args length,xRangeInt requires 1-3 int arguments"@H_301_4@)
    }
    var@H_301_4@ start,stop int@H_301_4@
    var@H_301_4@ step int@H_301_4@ = 1@H_301_4@
    switch@H_301_4@ len@H_301_4@(args) {
    case@H_301_4@ 1@H_301_4@:
        stop = args[0@H_301_4@]
        start = 0@H_301_4@
    case@H_301_4@ 2@H_301_4@:
        start,stop = args[0@H_301_4@],args[1@H_301_4@]
    case@H_301_4@ 3@H_301_4@:
        start,step = args[0@H_301_4@],args[1@H_301_4@],args[2@H_301_4@]
    }

    ch := make@H_301_4@(chan@H_301_4@ int@H_301_4@)
    go@H_301_4@ func@H_301_4@() {
        if@H_301_4@ step > 0@H_301_4@ {
            for@H_301_4@ start < stop {
                ch <- start
                start = start + step
            }
        } else@H_301_4@ {
            for@H_301_4@ start > stop {
                ch <- start
                start = start + step
            }
        }
        close@H_301_4@(ch)
    }()

    return@H_301_4@ ch
}

使用xRangeInt:

func@H_301_4@ main() {
    for@H_301_4@ v := range@H_301_4@ xRangeInt(20@H_301_4@, 10@H_301_4@, -3@H_301_4@) {
        fmt.Println(v)
    }
}
原文链接:https://www.f2er.com/go/188270.html

猜你在找的Go相关文章