Python-推导式

一、列表推导式

作用：用一个表达式创建一个有规律的列表或控制一个有规律的列表。列表推导式又叫列表生成式(简化代码)

1.1 示例

• while循环实现

  list1 = [] 
  i = 0 
  while i < 10: 
      list1.append(i) 
      i += 1 
  print(list1)
  

• 列表推导式

  list3 = [i for i in range(10)] 
  print(list3)
  

1.2 带if的列表推导式

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

• 通过range实现

list1 = [i for i in range(0,10,2)]
print(list)

• if条件实现

list = []
for i in range(10):
    if i % 2 == 0:
        list1.append(i)
print(list1)

# 推导式实现 
list2 = [i for i in range(10) if i % 2 == 0]
print(list2)

1.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)

# 推导式实现
list2 = [(i,j) for i in range(1,3) for j in range(3)]
print(list2)

二、字典推导式

思考：如果有如下两个列表：

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

如何快速合并为一个列表
字典推导式的作用：快速合并列表为字典中目标数据

2.1 示例

• 创建一个字典：字典key是1-5数字，value是这个数字的2次方

dict1 = {i: i**2 for i in range(1, 5)} 
print(dict1) # {1:1, 2: 4, 3: 9, 4: 16}

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

  list1 = ['name', 'age', 'gender', 'id'] 
  list2 = ['Tom', 20, 'man'] 
  dict1 = {} 
  for i in range(len(list1)): 
      dict1[list1[i]] = list2[i] 
  print(dict1) 
  
  # 使用推导式实现
  dict2 = {list1[i]: list2[i] for i in range(len(list1))} 
  print(dict2)
  

  总结：

  1. 如果两个列表数据个数相同，len统计任意一个列表的长度都可以；
  2. 如果两个列表数据个数不同，len统计数据列表少的列表即可。

• 提取字典当中的目标数据

  counts = {'MBP': 268, 'HP': 125, 'DELL': 201, 'Lenovo': 199, 'acer': 99} 
  count1 = {} 
  for key,value in counts.items(): 
        if value >= 200: 
            count1[key] = value 
  print(count1) 
  
  # 推导式实现
  count2 = {key: value for key, value in counts.items() if value >= 200} 
  print(count2)
  

三、集合推导式

需求：创建一个集合，数据为下方列表的2次方
list1 = [1,1,2]
代码如下:

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

集合有数据去重功能