目录
1.列表解析
2.生成器表达式
3.集合解析式
4.字典解析式
1. 列表解析介绍
1.1 列表解析介绍
语法:
[返回值 for 元素 in 可迭代对象 if 条件]
使用中括号[],内部是for循环,if条件语句可选
返回一个新列表
>>> even = []
>>> for x in range(10):
... if x % 2 == 0:
... even.append(x)
>>> print(even)
[0, 2, 4, 6, 8]
列表解析方法:
>>> even2 = [x for x in range(10) if x % 2 == 0]
>>> print(even2)
[0, 2, 4, 6, 8]
1.2 列表解析进阶
[expr for item in iterable if cond1 if cond2]
等价于
ret = []
for item in iterable:
if cond1:
if cond2:
ret.append(expr)
例1:20以内,既能被2整除又能被3整除的数
>>> [i for i in range(20) if i%2 == 0 and i%3 == 0]
[0, 6, 12, 18]
>>> [i for i in range(20) if i%2 == 0 if i%3 == 0]
[0, 6, 12, 18]
[expr for i in iterable1 for j in iterable2]
等价于
ret = []
for i in iterable1:
for j in iterable2:
ret.append(expr)
例2:
>>> [(x,y) for x in 'abcde' for y in range(2)]
[('a', 0), ('a', 1), ('b', 0), ('b', 1), ('c', 0), ('c', 1), ('d', 0), ('d', 1), ('e', 0), ('e', 1)]
>>> [[x,y] for x in 'abcde' for y in range(2)]
[['a', 0], ['a', 1], ['b', 0], ['b', 1], ['c', 0], ['c', 1], ['d', 0], ['d', 1], ['e', 0], ['e', 1]]
>>> [{x,y} for x in 'abcde' for y in range(2)]
[{0, 'a'}, {'a', 1}, {0, 'b'}, {1, 'b'}, {0, 'c'}, {1, 'c'}, {'d', 0}, {'d', 1}, {0, 'e'}, {'e', 1}]
2.生成器表达式
2.1 介绍
语法介绍:
(返回值 for 元素 in 可迭代对象 if 条件)
列表解析式的中括号换成小括号即可,返回一个生成器
列表解析的区别:
生成器表达式是按需计算(或称惰性求值,延迟计算),需要的时候才计算值列,表解析式是立即返回值
可迭代对象和迭代器均属于生成器
例1:
>>> g = ["{:04}".format(i) for i in range(1,5)]
>>> g
['0001', '0002', '0003', '0004']
>>> for x in g :
... print(x)
0001
0002
0003
0004
>>> for x in g:
... print(x)
0001
0002
0003
0004
使用生成器:
>>> g = ("{:04}".format(i) for i in range(1,5))
>>> g
<generator object <genexpr> at 0x7f1054aa3c80>
>>> next(g)
'0001'
>>> for x in g:
... print(x)
0002
0003
0004
>>> for x in g:
... print(x)
总结
生成器: 列表
1.延迟计算 立即计算
2.返回迭代器,可以迭代 返回的不是迭代器,返回可迭代对象列表
3.从前到后走完一遍,数据只能使用一次 数据可反复,多次使用
3.集合解析式
3.1 语法介绍
{返回值 for 元素 in 可迭代对象 if 条件}
列表解析式的中括号换成大括号{}就行了
立即返回一个集合
3.2 举例
>>> {(x,x+1) for x in range(3)}
{(0, 1), (1, 2), (2, 3)}
>>> {x for x in range(3)}
{0, 1, 2}
>>> {[x] for x in range(3)}
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "<stdin>", line 1, in <setcomp>
TypeError: unhashable type: 'list'
4.字典解析式
4.1 语法介绍
{返回值 for 元素 in 可迭代对象 if 条件}
列表解析式的中括号换成大括号{}就行了
使用key:value形式
立即返回一个字典
4.2 举例
>>> {x:x+1 for x in range(3)}
{0: 1, 1: 2, 2: 3}
>>> {x:[x,x+1] for x in range(3)}
{0: [0, 1], 1: [1, 2], 2: [2, 3]}
>>> {(x,):[x,x+1] for x in range(3)}
{(0,): [0, 1], (1,): [1, 2], (2,): [2, 3]}
>>> {chr(0x41+x):x**2 for x in range(3)}
{'A': 0, 'B': 1, 'C': 4}
>>> {str(x):y for x in range(3) for y in range(4)}
{'0': 3, '1': 3, '2': 3}
>>> {[x]:[x,x+1] for x in range(3)}
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "<stdin>", line 1, in <dictcomp>
TypeError: unhashable type: 'list'
网友评论