Python 语言有一种独特的推导式语法,相当于语法糖的存在,可以帮你在某些场合写出比较精简酷炫的代码。但没有它,也不会有太多的影响。Python 语言有几种不同类型的推导式,下面逐一介绍:
1. 列表推导式
列表推导式是一种快速生成列表的方式。其形式是用方括号括起来的一段语句,如下例子所示:
lis = [x * x for x in range(1, 10)]
print(lis)
------------------------------------
结果:[1, 4, 9, 16, 25, 36, 49, 64, 81]
列表推导式要这么理解,首先执行 for 循环,对于每一个 x,代入 x*x 中进行运算,将运算结果逐一添加到一个新列表内,循环结束,得到最终列表。它相当于下面的代码:
lis = []
for i in range(1, 10):
lis.append(i*i)
print(lis)
列表推导式为我们提供了一种在一行内实现较为复杂逻辑的生成列表的方法。其核心语法是用中括号[]将生成逻辑封装起来。
列表推导式有多种花样用法:
增加条件语句
>>> [x * x for x in range(1, 11) if x % 2 == 0]
[4, 16, 36, 64, 100]
通过在后面添加 if 子句,对 x 进行过滤。
多重循环
>>> [a + b for a in ‘123' for b in ‘abc']
['1a', '1b', '1c', '2a', '2b', '2c', '3a', '3b', '3c']
同时循环 a 和 b 两个变量。 2. 字典推导式
既然使用中括号[]可以编写列表推导式,那么使用大括号呢?使用大括号{}可以制造字典推导式!
>>> dic = {x: x**2 for x in (2, 4, 6)}
>>> dic
{2: 4, 4: 16, 6: 36}
>>> type(dic)
<class 'dict'>
注意 x: x**2 的写法,中间的冒号,表示左边的是 key 右边的是 value。
3. 集合推导式
大括号除了能用作字典推导式,还可以用作集合推导式,两者仅仅在细微处有差别。
>>> a = {x for x in 'abracadabra' if x not in 'abc'}
>>> a
{'d', 'r'}
>>> type(a)
<class 'set'>