列表生成式
生成列表可以用列表生成式的方法
#使用列表生成式
>>>range(1,11)
range(1, 11)
>>>[x*x for x in range(1,11)]
[1,4, 9, 16, 25, 36, 49, 64, 81, 100]
#使用两层循环
>>>[m+n for m in 'abc' for n in '123']
['a1', 'a2', 'a3', 'b1','b2', 'b3', 'c1', 'c2', 'c3']
#大小写转换
>>>L = ['Hello', 'World', 'IBM', 'Apple']
>>>[s.lower() for s in L]
['hello', 'world', 'ibm','apple']
生成器
通过列表生成式,我们可以直接创建一个列表。但是,受到内存限制,列表容量肯定是有限的。在Python中,这种一边循环一边计算的机制,称为生成器(Generator
)。要创建一个generator,有很多种方法。第一种方法很简单,只要把一个列表生成式的[]
改成()
,就创建了一个generator:
>>> g=(x*x for x in range(1,11))
>>> g
<generatorobject <genexpr> at 0x0289AB98>
#generator是可迭代对象,使用for循环直接打印出list的每一个元素
>>>for i in g:
print(i)
1
4
9
16
25
36
49
64
81
100
#生成斐波那契数列
>>> def fib(max):
n, a, b = 0, 0, 1
while n < max:
print (b)
a, b = b, a + b
n = n + 1
>>> fib(10)
1
1
2
3
5
8
13
21
34
55
generator和函数的执行流程不一样。函数是顺序执行,遇到return语句或者最后一行函数语句就返回。而变成generator的函数,在每次调用next()
的时候执行,遇到yield
语句返回,再次执行时从上次返回的yield
语句处继续执行。要理解generator的工作原理,它是在for
循环的过程中不断计算出下一个元素,并在适当的条件结束for
循环。对于函数改成的generator来说,遇到return语句或者执行到函数体最后一行语句,就是结束generator的指令,for
循环随之结束。