Python - 推导式 - 进阶之路 - Day03

推导式comprehensions（又称解析式），是Python的一种独有特性。推导式是可以从一个数据序列构建另一个新的数据序列的结构体。 共有三种推导，在Python2和3中都有支持：

• 列表(list)推导式
• 字典(dict)推导式
• 集合(set)推导式

一、列表推导式

列表推导式的基本格式

new_list = [expression for_loop_expression if condition]

1. 先举个例子，我想找出一个数值列表中为偶数的元素，并组成新列表，通常不用列表推导式，可以这么写

old_list = [0,1,2,3,4,5]

new_list = []
for item in old_list:
    if item % 2 == 0:
        new_list.append(item)

print(new_list) 

运行结果：

[0, 2, 4]

一个简单的功能，写的代码倒是不少。
如果使用了列表推导式，那就简洁多了，而且代码还变得更加易读了。

old_list = [0,1,2,3,4,5]

new_list = [item for item in old_list if item % 2 == 0]
print(new_list)

运行结果：

[0, 2, 4]

2. 再举个例子，求(x,y),其中x是0-5之间的偶数，y是0-5之间的奇数组成的元祖列表:

list = [(x,y) for x in range(5) if x%2 == 0 for y in range(5) if y%2 == 1]
print(list)

运行结果：

[(0, 1), (0, 3), (2, 1), (2, 3), (4, 1), (4, 3)]

二、字典推导式

字典推导式的基本格式，和 列表推导式相似，只是把 [] 改成了 {}，并且组成元素有两个：key 和 value，要用 key_expr: value_expr 表示。

new_dict ={ key_expr: value_expr for_loop_expression if condition }

1. 举个例子，我想从一个包含所有学生成绩信息的字典中，找出数学考满分的同学。

old_student_score_info = {
    "Jack": {
        "chinese": 87,
        "math": 92,
        "english": 78
    },
    "Tom": {
        "chinese": 92,
        "math": 100,
        "english": 89
    }
}


new_student_score_info = {}
for name, scores in old_student_score_info.items():
    if scores["math"] == 100:
        new_student_score_info.setdefault(name, scores)
print(new_student_score_info)

而用字典推导式，一行代码即可做到： 

old_student_score_info = {
    "Jack": {
        "chinese": 87,
        "math": 92,
        "english": 78
    },
    "Tom": {
        "chinese": 92,
        "math": 100,
        "english": 89
    }
}

new_student_score_info = {name: scores for name, scores in old_student_score_info.items() if scores["math"] == 100}
print(new_student_score_info)

运行结果：

{'Tom': {'chinese': 92, 'math': 100, 'english': 89}}

2. 再举一个简单的例子，我们有一个fruit的list，现在想要得到每一种水果的单词长度: 

fruits = ['apple', 'orange', 'banana', 'mango', 'peach']

fruits_dict = {}
for fruit in fruits:
    fruits_dict.setdefault(fruit, len(fruit))
print(fruits_dict)

 而字典推导式则更简单

fruits = ['apple', 'orange', 'banana', 'mango', 'peach']

fruits_dict = {fruit: len(fruit) for fruit in fruits}
print(fruits_dict)

运行结果：

{'apple': 5, 'orange': 6, 'banana': 6, 'mango': 5, 'peach': 5}

三、集合推导式 

集合推导式跟列表推导式也是类似的。唯一的区别在于它使用大括号{}，组成元素也只要一个。
基本格式

new_set = { expr for_loop_expression if condition }

 1. 举个例子，我想把一个数值列表里的数进行去重处理

old_list = [0, 0, 0, 1, 2, 3]

new_set = {item for item in old_list}
print(new_set)

运行结果：

{0, 1, 2, 3}

以上简单的列举了几个例子，大家也可以参考以下几个文档

Python中的推导式_鸿神的博客-CSDN博客

Python推导式的五种写法 - 知乎

python各种推导式（超级详细） - 简书