列表生成式
列表生成式是python内置的非常简单缺强大的可以用来创建list的生成式
生成一个[1*1, 2*2, 3*3, 4*4, 5*5, 6*6, 7*7, 8*8, 9*9, 10*10]
>>> [x * x for x in range(1, 11)]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
写列表生成式时, 把生成的元素放到最前面,后面跟for
循环,就可以把list创建出来。当然,在for循环后面还可以加上if判断,可以做到筛选的功能,即,取偶数的平方
>>> [x ** 2 x in range(1, 11) if x % 2 == 0]
[4, 16, 36, 64, 100]
循环嵌套
>>> [m + n for m in 'ABC' for n in 'XYZ']
['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']
运用列表生成式,可以写出非常简洁的代码。如:列出当前目录下的所有文件和目录
>>> import os
>>> [d for d in os.listdir('.')]
['.emacs.d', '.ssh', '.Trash', 'Adlm', 'Applications', 'Desktop', 'Documents',
'Downloads', 'Library', 'Movies', 'Music', 'Pictures', 'Public', 'VirtualBox VMs', 'Workspace', 'XCode']
for循环可以同事使用多个变量,如dict
的items()
可以i同时迭代key和value:
>>> d = {'x': 'A', 'y': 'B', 'z': 'C'}
>>> for k, v in d.items():
print(k, '=', v)
y = B
x = A
z = C
# 列表生成式
>>> [k + '=' + v for k, v in d.items()]
['y=B', 'x=A', 'z=C']
生成器generator
在循环中不断推算后续的元素,一边循环一遍计算的机制,称为生成器:generator。
generator和list两种生成器的区别在于最外层的[]
和()
,list可以直接打印出每一个元素,但打印generator的元素,可以通过next()
函数获得generator的下一个返回值。
如果一个函数函数定义中包含yield
关键字,这个函数就不再是一个普通函数,而是一个generator。
generatir和函数的执行流程不一样。函数是顺序执行,遇到return
或者最后一行函数语句就返回,当变成generator的函数,在每次调用next()
的时候执行,遇到yield
语句返回,再次执行从上次返回的yield
语句处继续执行。
迭代器Iterator
凡是可以作用于for
循环的对象都是Iterable
类型
凡是可作用于next()
函数的对象都是Iterator
类型,表示一个惰性计算的序列;
几何数据类型如list
,dict
,str
等是Iterable
但不是Iterator
,不过可以通过iter()
函数获得一个Iterator
对象。