for循环用腻了,试试列表生成式。
- 2019 年 10 月 10 日
- 笔记
在编写程序或者查看别人的程序时,经常会遇到列表生成式,这个使用起来并不复杂,但是非常有用,使我们的代码更加简洁灵活。很多python使用者并不太会使用它。今天,就给大家详细讲解列表生成式和生成器表达式的使用
一、列表生成式
优点:一行代码几乎可以搞定所需要的任何列表
缺点:容易着迷,不易排错,不能超过三个循环
- 用列表推导式能构建的任何列表,用别的都可以构建,比如for循环
- 列表推导式,最多不超过3个for循环。判断只能用一个
1. 循环模式:模板:[经过加工的i for i in 可迭代对象]
使用for循环方法
li = [] for i in range(1,4): li.append('还珠格格第'+str(i)+'部') print(li) #['还珠格格第1部', '还珠格格第2部', '还珠格格第3部']
第二种写法
li = [] for i in range(1,4): li.append('还珠格格第%s部' % i) print(li)
上面的代码,可以一行搞定。用列表推导式就可以了
li = ['还珠格格第%s部' %i for i in range(1,4)] print(li) li = ['还珠格格第{}部'.format(i) for i in range(1,4)] # 上下结果相同 print(li)
例:求1~10平方结果
li = [i ** 2 for i in range(1,11)] print(li)
执行输出:[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
2. 筛选模式:[经过加工的i for i in 可迭代对象 if 条件 筛选]
30以内所有能被3整除的数
l3 = [i for i in range(1,31) if i % 3 == 0] print(l3) # [3, 6, 9, 12, 15, 18, 21, 24, 27, 30]
30以内所有能被3整除的数的平方
li = [i**2 for i in range(1,31) if i % 3 == 0] print(li) # [9, 36, 81, 144, 225, 324, 441, 576, 729, 900]
找到嵌套列表中名字含有两个'e'的所有名字
names = [['Tom', 'Billy', 'Jefferson', 'Andrew', 'Wesley', 'Steven', 'Joe'],['Alice', 'Jill', 'Ana', 'Wendy', 'Jennifer', 'Sherry']] l4 = [name for i in names for name in i if name.count('e') == 2 ] print(l4) # ['Jefferson', 'Wesley', 'Steven', 'Jennifer']
常用的是列表推导式
字典推导式
将一个字典的key和value对调
dict = {'a': 10, 'b': 34} dict_frequency = {dict[k]: k for k in mcase} print(dict_frequency)
执行输出:{10: 'a', 34: 'b'}
相当于
dict = {'a': 10, 'b': 34} dict_frequency = {} for k in dict: dict_frequency[k]=dict[k] print(dict_frequency)
如果Key和value是一样的,不适合上面的代码
集合推导式
计算列表中每个值的平方,自带去重功能
squared = {x**2 for x in [1, -1, 2]} print(squared)
执行输出:{1, 4}
a = {1,1,4} print(type(a))
执行输出:<class 'set'> 结果是一个集合,它也是用{}表示的。
集合和字典是有区别的:
有键值对的,是字典,比如{‘k1′:1,’k1’:2}
没有键值对的,是集合,比如{1,2,3,4}
二、生成器表达式
与列表推导式相同,只是将[ ]改为( )
l_obj = ('还珠格格第%s部' % i for i in range(1,4)) print(l_obj)
结果是一个生成器对象:<generator object <genexpr> at 0x000002DDBEBADE60>
取值使用__next__方法
l_obj = ('还珠格格第%s部' % i for i in range(1,4)) print(l_obj.__next__()) print(l_obj.__next__()) print(l_obj.__next__())
输出结果: 还珠格格第1部 还珠格格第2部 还珠格格第3部
列表推导式:一目了然,占内存
生成器表达式: 不便看出,节省内存。