推导式(comprehensions),又称解析式。利用列表推导式、字典推导式、集合推导式可以从一个数据对象构建另一个新的数据对象。利用生成器推导式可以构建生成器对象。
列表推导式(list comprehension)是Python开发时用得最多的技术之一,表示对可迭代(Iterable)对象的元素进行遍历、过滤或再次计算,生成满足条件的新列表。它的结构是在一个方括号里包含一个函数或表达式(再次计算),接着是一个for语句(遍历),然后是0个或多个for(遍历)或者if语句(过滤),在逻辑上等价于循环语句,但是形式上更简洁。
语法形式:
[function / expression for value1 in Iterable1 if condition1
for value2 in Iterable2 if condition2
……
for valuen in Iterablen if conditionn ]
1.列表推导式和循环语句for
2.列表推导式和条件语句if
在列表推导式中,条件语句if对可迭代(Iterable)对象中的元素进行筛选,起到过滤的作用。
>>> n=[10,-33,21,5,-7,-9,3,28,-16,37]
>>> number=[i*2 for i in n if i>0]
>>> number
[20, 42, 10, 6, 56, 74]
在列表推导式中使用if else语句
>>> n=[10,-33,21,5,-7,-9,3,28,-16,37]
>>> number=[i*2 if i%2==0 else i*3 for i in n if i>0]
>>> number
[20, 63, 15, 9, 56, 111]
3.列表推导式和函数function
在列表推导式中可以使用函数function
>>> import random
>>> rlist=[random.randint(30,80) for i in range(15)] #使用randint()函数
>>> rlist
[30, 45, 68, 60, 73, 49, 66, 72, 66, 33, 52, 52, 61, 42, 51]
>>> def ff(x):
if x%3==0:
x/=3
elif x%7==0:
x*=2
else:
x*=5
return x
>>> number=[ff(i) for i in rlist] #使用自定义的ff()函数
>>> number
[10.0, 15.0, 340, 20.0, 365, 98, 22.0, 24.0, 22.0, 11.0, 260, 260, 305, 14.0, 17.0]
4.同时遍历多个列表或可迭代对象
>>> score1=[86,78,98,90,47,80,90]
>>> score2=[87,78,89,92,90,47,85]
>>> nn=[(i,j) for i in score1 if i>=90 for j in score2 if j<=85]
>>> nn
[(98, 78), (98, 47), (98, 85), (90, 78), (90, 47), (90, 85), (90, 78), (90, 47), (90, 85)]
字典推导式和列表推导式的使用方法类似,只不过将方括号变成花括号,并且需要两个表达式,一个生成键,一个生成值,两个表达式之间使用冒号分隔,最后生成的是字典。
语法形式:
{key_expression: value_expression for value1 in Iterable1 if condition1]
for value2 in Iterable2 if condition2
……
for valuen in Iterablen if condition}
例如,列表name存储若干人的名字(唯一),列表score在对应的位置上存储这些人的成绩,利用字典推导式,以名字为键、成绩为值组成新字典dd。
>>> name= ['Bob','Tom','Alice','Jerry','Wendy','Smith']
>>> score=[86,78,98,90,47,80]
>>> dd={i:j for i,j in zip(name,score)}
>>> dd
{'Bob': 86, 'Tom': 78, 'Alice': 98, 'Jerry': 90, 'Wendy': 47, 'Smith': 80}
以名字为键、成绩为值组成新字典exdd,新字典中的键值对只包含成绩80及以上的。
>>> exdd={i:j for i,j in zip(name,score) if j>=80}
>>> exdd
{'Bob': 86, 'Alice': 98, 'Jerry': 90, 'Smith': 80}
在上面生成的字典dd中挑出成绩及格的组成新字典pdd。
>>> pdd={i:j for i,j in dd.items() if j>=60}
>>> pdd
{'Bob': 86, 'Tom': 78, 'Alice': 98, 'Jerry': 90, 'Smith': 80}
以名字为键、名字的长度为值组成新字典nd
>>> nd={i:len(i) for i in name}
>>> nd
{'Bob': 3, 'Tom': 3, 'Alice': 5, 'Jerry': 5, 'Wendy': 5, 'Smith': 5}
集合也有自己的推导式,跟列表推导式类似,只不过将方括号变成花括号,最后生成的是集合。
语法形式:
{function / expression for value1 in Iterable1 if condition1
for value2 in Iterable2 if condition2
……
for valuen in Iterablen if conditionn }
生成器推导式用法与列表推导式类似,把列表推导式的方括号改成圆括号。它与列表推导式最大的区别是:生成器推导式的结果是一个生成器对象,是一种迭代器(Iterator);而列表推导式的结果是一个列表。
生成器对象可以通过for循环或者__next__()方法、next()函数进行遍历,也可以转换为列表或元组,但是不支持使用下标访问元素,已经访问过的元素也不支持再次访问。当所有元素访问结束之后,如果想再次访问就必须重新创建该生成器对象。
>>> gen=(int(i/3) for i in range(1,10) if i%3==0)
>>> gen
<generator object <genexpr> at 0x0000000002F0BB88>
>>> list(gen) #生成器对象转换为列表
[1, 2, 3]
>>> gen.__next__() #不能再次访问
Traceback (most recent call last):
File "" , line 1, in <module>
gen.__next__()
StopIteration
>>> gen=(int(i/3) for i in range(1,10) if i%3==0) #重新生成生成器对象
>>> gen.__next__() #访问下一个元素
1
>>> next(gen) #访问下一个元素
2
>>> gen.__next__()
3
>>> next(gen) #访问完毕,不能再次访问
Traceback (most recent call last):
File "" , line 1, in <module>
next(gen)
StopIteration
>>> gen=(int(i/3) for i in range(1,10) if i%3==0)
>>> for i in gen: #for循环遍历
print(i,end=' ')
1 2 3