列表生成式
列表生成式即List Comprehensions:
是Python内置的非常简单却强大的可以用来创建list的生成式。
列表生成式案例1
要生成list [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]:
可以用list(range(1, 11)):
list(range(1, 11)) 复制代码
结果为:[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
但如果要生成[1x1, 2x2, 3x3, ..., 10x10]怎么做?
方法一是循环:
L = [] for x in range(1, 11): L.append(x * x) 复制代码
结果为:
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
但是循环太繁琐,
而列表生成式则可以用一行语句代替循环生成上面的list:
[x * x for x in range(1, 11)] 复制代码
结果为:[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
写列表生成式时,把要生成的元素x * x放到前面,后面跟for循环,就可以把list创建出来,简单、方便。
列表生成式案例2
for循环后面还可以加上if判断:
这样我们就可以筛选出仅偶数的平方:
[x * x for 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']
列表生成式案例3
运用列表生成式,可以写出非常简洁的代码。
例如,列出当前目录下的所有文件和目录名,可以通过一行代码实现:
注意:os.listdir可以列出文件和目录
import os # 导入os模块,模块的概念后面讲到 [d for d in os.listdir('.')] 复制代码
列表生成式案例4
for循环其实可以同时使用两个甚至多个变量,比如dict的items()可以同时迭代key和value:
d = {'x': 'A', 'y': 'B', 'z': 'C' } for k, v in d.items(): print(k, '=', v) 复制代码
结果为:y = B x = A z = C
因此,列表生成式也可以使用两个变量来生成list:
d = {'x': 'A', 'y': 'B', 'z': 'C' } [k + '=' + v for k, v in d.items()] 复制代码
结果为:['y=B', 'x=A', 'z=C']
列表生成式案例5
把一个list中所有的字符串变成小写:
L = ['Hello', 'World', 'IBM', 'Apple'] [s.lower() for s in L] 复制代码
结果为:['hello', 'world', 'ibm', 'apple']
迭代器(Iterator)
可以直接作用于for循环的数据类型有以下几种:
一类是集合数据类型,如list、tuple、dict、set、str等;一类是generator,包括生成器和带yield的generator function。
可迭代对象:Iterable
这些可以直接作用于for循环的对象统称为可迭代对象:Iterable。
可以使用isinstance()判断一个对象是否是Iterable对象:
而生成器不但可以作用于for循环,还可以被next()函数不断调用并返回下一个值,直到最后抛出StopIteration错误表示无法继续返回下一个值了。
迭代器:Iterator
可以被next()函数调用并不断返回下一个值的对象称为迭代器:Iterator。
可以使用isinstance()判断一个对象是否是Iterator对象,list、dict、str虽然是Iterable,却不是Iterator。
iter()函数
把list、dict、str等Iterable变成Iterator可以使用iter()函数:
isinstance(iter('abc'), Iterator) 复制代码
结果为:True
Iterator甚至可以表示一个无限大的数据流,例如全体自然数。而使用list是永远不可能存储全体自然数的。
小结
凡是可作用于for循环的对象都是Iterable类型;凡是可作用于next()函数的对象都是Iterator类型,它们表示一个惰性计算的序列;集合数据类型如list、dict、str等是Iterable,但不是Iterator,不过可以通过iter()函数获得一个Iterator对象。
作者:zhulin1028
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。