Python函数和常用模块-生成器 | 【韩涛博客】

Python函数和常用模块-生成器

Python函数和常用模块-生成器

列表生成式:

[i*2 for i in range(10)]

生成[0,2,4,6,8,10,12,14,16,18]

注:列表生成式可以直接生成列表,但是当数据量特别大的时候,会特别占内存,而且不需要的内容也会占用到空间。

生成器:节省内存空间,只有在调用到的时候才会生成数据

节省内存方式:只记录当前的数据位置,之前的数据会释放。只有一个next方法:_next_()

要创建一个generator,有很多种方法。第一种方法只要把一个列表生成式的[]改成(),就创建了一个generator:

eg:

a=[ i * i for  i in range(10)]
print(a)
b=( i * i for  i in range(10))
print(b)
输出:
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
<generator object <genexpr> at 0x02126240>

eg:输出斐波那契数列的前N个数:

函数方法:

def fib(max):
    n, a, b = 0, 0, 1
    while n < max:
        print(b)
        a, b = b, a + b
        '''t = (b, a + b) # t是一个tuple
        a = t[0]
        b = t[1]'''
        n = n + 1
    return 'done'
fib(5)

生成器方法:

def fib(max):
    n,a,b = 0,0,1
    while n < max:
        #print(b)
        yield  b
        a,b = b,a+b
        n += 1
    return '-----done-----'#功能是异常时打印消息
f=fib(5)
print(f)
输出:<generator object fib at 0x02126210>
print(f.__next__()) #逐行执行并打印
print(f.__next__())
print("干点别的事")
print(f.__next__())
输出:
1
1
干点别的事
2

注:

1.generator和函数的执行流程不一样。函数是顺序执行,遇到return语句或者最后一行函数语句就返回。而变成generator的函数,在每次调用next()的时候执行,遇到yield语句返回,再次执行时从上次返回的yield语句处继续执行。

2.在循环过程中不断调用yield,就会不断中断。把函数改成generator后,我们基本上不会用next()来获取下一个返回值,而是直接使用for循环:

for n in fib(6):
    print(n)

用for循环调用generator时,发现拿不到generator的return语句的返回值。如果想要拿到返回值,必须捕获StopIteration错误,返回值包含在StopIteration的value中:

def fib(max):
    n,a,b = 0,0,1
    while n < max:
        #print(b)
        yield  b
        a,b = b,a+b
        n += 1
    return '-----done-----'
f=fib(5)
g = fib(6)
while True:
    try:
        x = next(g)
        print('g:', x)
    except StopIteration as e:
        print('Generator return value:', e.value)
        break
输出:
g: 1
g: 1
g: 2
g: 3
g: 5
g: 8
Generator return value: -----done-----
广告

发表评论 取消回复
表情 图片 链接 代码

分享
×
订阅图标按钮