python基础笔记——推导式（生成式）

范围：

• 列表推导式

• 字典推导式

• 集合推导式

一、列表推导式

作用：用一个表达式创建一个有规律的列表或控制一个有规律的列表

1、快速体验

需求：创建一个0-10的列表

list1 = []

# 1.while循环实现
i = 0
while i < 10:
    list1.append(i)
    i += 1
print(list1)  # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

# 2.for循环实现
list2 = []
for i in range(10):
    list2.append(i)
print(list2)  # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

# 3.列表推导式实现
# 第一个i代表append的i
list3 = [i for i in range(10)]
print(list3)  # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

2、带if的列表推导式

需求：创建0-10的偶数列表

# 1.简单列表推导式 range步长
list1 = [i for i in range(0, 10, 2)]
print(list1)  # [0, 2, 4, 6, 8]

# 2.for循环加if 创建有规律的列表
list2 = []
for i in range(10):
    if i % 2 == 0:
        list2.append(i)
print(list2)  # [0, 2, 4, 6, 8]

# 3.把for循环配合if的代码改写为带if的列表推导式
list3 = [i for i in range(10) if i % 2 == 0]
print(list3)  # [0, 2, 4, 6, 8]

3、多个for循环实现列表推导式

需求：创建列表如下：

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

list1 = []
for i in range(1, 3):
    for j in range(3):
        # 列表里面追加元组：循环前准备空列表，然后在这里追加元组数据到列表
        list1.append((i, j))
print(list1)  # [(1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]

# 多个for实现列表推导式
list2 = [(i, j) for i in range(1, 3) for j in range(3)]
print(list2)  # [(1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]

# 其实多for的列表推导式等同于for嵌套循环

二、字典推导式

作用：快速合并列表为字典或提取字典中目标数据

1、快速体验

"""
需求：创建一个字典：字典的key是1-3数字，value是这个数字的2次方
"""
dict1 = {i: i**2 for i in range(1, 4)}
print(dict1)  # {1: 1, 2: 4, 3: 9}

将两个列表合并为一个字典：

list1 = ['name', 'age', 'gender', 'id']
list2 = ['Tom', 20, 'man']

dict1 = {list1[i]: list2[i] for i in range(len(list1))}
print(dict1)  # {'name': 'Tom', 'age': 20, 'gender': 'man'}

# 总结：
# 1.如果两个列表数据个数相同，len统计任何一个列表的长度都可以
# 2.如果两个列表数据个数不同，len统计数据少的列表数据个数

提取字典中目标数据：

# 需求：提取电脑台数大于等于200
counts = {'MBP': 221, 'HP': 123, 'DELL': 202, 'lenovo': 199}

dict1 = {key: value for key, value in counts.items() if value >= 200}
print(dict1)  # {'MBP': 221, 'DELL': 202}

三、集合推导式

list1 = [1, 1, 2]

set1 = {i ** 2 for i in list1}
print(set1)  # {1, 4}