【Python】推导式_列表推导式

推导式（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]
1
2
3
4
5

在列表推导式中使用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]
1
2
3
4

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]
1
2
3
4

>>> 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]
1
2
3
4
5
6
7
8
9
10
11
12

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)]
1
2
3
4
5

字典推导式

字典推导式和列表推导式的使用方法类似，只不过将方括号变成花括号，并且需要两个表达式，一个生成键，一个生成值，两个表达式之间使用冒号分隔，最后生成的是字典。
语法形式：
{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}
1
2
3
4
5

以名字为键、成绩为值组成新字典exdd，新字典中的键值对只包含成绩80及以上的。

>>> exdd={i:j for i,j in zip(name,score) if j>=80}
>>> exdd
{'Bob': 86, 'Alice': 98, 'Jerry': 90, 'Smith': 80}
1
2
3

在上面生成的字典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}
1
2
3

以名字为键、名字的长度为值组成新字典nd

>>> nd={i:len(i) for i in name}
>>> nd
{'Bob': 3, 'Tom': 3, 'Alice': 5, 'Jerry': 5, 'Wendy': 5, 'Smith': 5}
1
2
3

集合推导式

集合也有自己的推导式，跟列表推导式类似，只不过将方括号变成花括号，最后生成的是集合。
语法形式：
{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 "<pyshell#29>", 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 "<pyshell#34>", 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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30