golang实现python xrange方法

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

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

输出:

0
1
2
3
4

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

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

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

    return ch
}

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

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

输出:

10
12
14
16
18

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

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

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

    return ch
}

使用xRangeInt:

func main() {
    for v := range xRangeInt(20, 10, -3) {
        fmt.Println(v)
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值