基于python生成器重写range函数

      众所周知,在python中range函数非常的好用,试想我们自己是否可以写一个类似range 函数的函数呢,而从根本上range相当于一个生成器(但不能被next()调用),刚学完生成器的我,决定写一个_range(),用于生成两数之间的所有偶数,不说了上代码

def _range(i=0,n=0):
    '''
    这是我写的类range函数,用于迭代i,n之间的所有偶数
    若n是偶数则不包括n,
    若i是偶数则包括i
    '''
    if n==0 and i>n:
        i,n=n,i
    while i<n:
        if i%2!=0:
            i=i+1
        if i ==n:
            break
        yield i
        i += 2

这里用到了yield关键字,用法跟range差不多,不过还可以ne呗xt()调用 ,例:

p=_range(10)
for i in range(5):
    print(next(p))

其他用法可以用range的地方都可以用_range

猜你喜欢

转载自blog.csdn.net/SEGeeK/article/details/81783156
今日推荐