[Python知识库]Python中的三种推导式及用法

Python中的推导式

推导式，也被称为解析式，是Python中一种独有的特性。
推导式是可以从一个数据序列构建另一个新的数据序列的结构体。
在Python中，共有三种推导式，分别是列表推导式、字典推导式和集合推导式。

列表推导式

列表推导式，也可以称为列表解析式，是创建一个新列表的快捷方式。
具体的结构为：在一个中括号[] 中包含一个表达式，然后紧跟一个for循环语句，再是0个或者多个for循环语句或者if语句，返回的结果将会是一个新列表。
中括号中包含的表达式是任意的，表示的是在列表中要存放的任意类型的对象。

下面是一个简单的例子帮助理解列表推导式。

nums = [i for i in range(30) if i%3 == 0]
print(nums)

输出：

[0, 3, 6, 9, 12, 15, 18, 21, 24, 27]

可以出看出，列表推导式可以快速地生成列表，特别是使用for循环来生成一个列表时，推荐使用列表推导式。

squares = []
for i in range(10):
    squares.append(i**2)

print(squares)

输出：

[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

上面的代码，可以用列表推导式来简化一下。

squares = [i**2 for i in range(10)]
print(squares)

输出：

[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

但需要注意的是，使用列表推导式来创建新列表时，应尽量保持代码的简短。如果列表推导式的代码超过了两行，为了代码的可读性，建议使用for循环。

字典推导式

字典推导式，也可以被称为字典解析式，是快速创建字典的快捷方式。
字典推导式可以从任何以键值对为元素的可迭代对象中创建新的字典。
字典推导式和列表推导式类似，但不再使用中括号，而是使用大括号{}，且存放的是键值对。

dial_code = [(86, 'China'), (91, 'India'),(7, 'Russia')]
country_code = {v:k for k, v in dial_code}
print(country_code)

输出：

{'China': 86, 'India': 91, 'Russia': 7}

使用字典推导式可以快速地将字典的键和值对调。

case = {'a': 12, 'A': 7}
new_case = {v:k for k, v in case.items()}
print(new_case)

输出：

{12: 'a', 7: 'A'}

集合推导式

集合推导式，也称为集合解析式，是快速创建集合的快捷方式。
集合推导式，与列表推导式比较，不再使用中括号，而是使用大括号{}；与字典推导式比较，都是使用大括号，但不存放键值对，而是存放任意类型的对象。

squares = {i**2 for i in range(5)}
print(squares)

输出：

{0, 1, 4, 9, 16}

集合推导式同样也自带去重复的功能。

squares = {i**2 for i in [1, 2, -1, 3, -2]}
print(squares)

输出：

{1, 4, 9}