列表推导式#

列表推导式格式为：

1
[表达式 for 变量 in 列表]
2
[out_exp_res for out_exp in input_list]
3

4
或者
5

6
[表达式 for 变量 in 列表 if 条件]
7
[out_exp_res for out_exp in input_list if condition]

• out_exp_res：列表生成元素表达式，可以是有返回值的函数。
• for out_exp in input_list：迭代 input_list 将 out_exp 传入到 out_exp_res 表达式中。
• if condition：条件语句，可以过滤列表中不符合条件的值。

过滤掉长度小于或等于3的字符串列表，并将剩下的转换成大写字母：

实例#

1
>>> names = ['Bob', 'Tom', 'alice', 'Jerry', 'Wendy', 'Smith']
2
>>> new_names = [name.upper() for name in names if len(name) > 3]
3
>>> print(new_names)
4
['ALICE', 'JERRY', 'WENDY', 'SMITH']

计算 30 以内可以被 3 整除的整数：

实例#

1
>>> multiples = [i for i in range(30) if i % 3 == 0]
2
>>> print(multiples)
3
[0, 3, 6, 9, 12, 15, 18, 21, 24, 27]

字典推导式#

字典推导基本格式：

1
{ key_expr: value_expr for value in collection }
2

3
或
4

5
{ key_expr: value_expr for value in collection if condition }

使用字符串及其长度创建字典：

实例#

1
listdemo = ['Google', 'Runoob', 'Taobao']
2
# 将列表中各字符串值为键，各字符串的长度为值，组成键值对
3
>>> newdict = {key: len(key) for key in listdemo}
4
>>> newdict
5
{'Google': 6, 'Runoob': 6, 'Taobao': 6}

提供三个数字，以三个数字为键，三个数字的平方为值来创建字典：

实例#

1
>>> dic = {x: x**2 for x in (2, 4, 6)}
2
>>> dic
3
{2: 4, 4: 16, 6: 36}
4
>>> type(dic)
5
<class 'dict'>

集合推导式#

集合推导式基本格式：

1
{ expression for item in Sequence }
2
或
3
{ expression for item in Sequence if conditional }

计算数字 1,2,3 的平方数：

实例#

1
>>> setnew = {i**2 for i in (1, 2, 3)}
2
>>> setnew
3
{1, 4, 9}

判断不是 abc 的字母并输出：

实例#

1
>>> a = {x for x in 'abracadabra' if x not in 'abc'}
2
>>> a
3
{'d', 'r'}
4
>>> type(a)
5
<class 'set'>

元组推导式（生成器表达式）#

元组推导式可以利用 range 区间、元组、列表、字典和集合等数据类型，快速生成一个满足指定需求的元组。

元组推导式基本格式：

1
(expression for item in Sequence )
2
或
3
(expression for item in Sequence if conditional )

元组推导式和列表推导式的用法也完全相同，只是元组推导式是用 () 圆括号将各部分括起来，而列表推导式用的是中括号 []，另外元组推导式返回的结果是一个生成器对象。

例如，我们可以使用下面的代码生成一个包含数字 1~9 的元组：

实例#

1
>>> a = (x for x in range(1, 10))
2
>>> a
3
<generator object <genexpr> at 0x7faf6ee20a50>  # 返回的是生成器对象
4

5
>>> tuple(a)  # 使用 tuple() 函数，可以直接将生成器对象转换成元组
6
(1, 2, 3, 4, 5, 6, 7, 8, 9)

练习题#

• 1. 列表推导式 创建文件 test_list.py，完成以下操作：

  • 用列表推导式生成 1-20 中所有偶数的列表
  • 用列表推导式将列表中的字符串转换为大写

  知识点： [表达式 for 变量 in 序列 if 条件]

  参考答案：

  1
  # 1-20 中所有偶数
  2
  even = [x for x in range(1, 21) if x % 2 == 0]
  3
  print(f"偶数列表: {even}")
  4
  
  5
  # 字符串转大写
  6
  words = ["hello", "world", "python"]
  7
  upper_words = [word.upper() for word in words]
  8
  print(f"大写列表: {upper_words}")

• 2. 字典推导式 创建文件 test_dict.py，完成以下操作：

  • 用字典推导式创建平方数字典 {1:1, 2:4, 3:9, 4:16, 5:25}
  • 用字典推导式将两个列表合并为字典

  知识点： {key: value for 变量 in 序列}

  参考答案：

  1
  # 平方数字典
  2
  squares = {x: x**2 for x in range(1, 6)}
  3
  print(f"平方数字典: {squares}")
  4
  
  5
  # 两个列表合并为字典
  6
  keys = ["name", "age", "city"]
  7
  values = ["张三", 25, "北京"]
  8
  d = {k: v for k, v in zip(keys, values)}
  9
  print(f"合并字典: {d}")

• 3. 集合推导式 创建文件 test_set.py，完成以下操作：

  • 用集合推导式去除列表中的重复元素
  • 用集合推导式从字符串中提取不重复的字符

  知识点： {表达式 for 变量 in 序列}

  参考答案：

  1
  # 去除重复元素
  2
  lst = [1, 2, 2, 3, 3, 3, 4, 4, 4, 4]
  3
  unique = {x for x in lst}
  4
  print(f"去重后: {unique}")
  5
  
  6
  # 提取不重复字符
  7
  s = "abracadabra"
  8
  chars = {x for x in s}
  9
  print(f"不重复字符: {chars}")

• 4. 元组推导式 创建文件 test_tuple.py，完成以下操作：

  • 用元组推导式生成 1-10 的平方数
  • 将生成器转换为元组

  知识点： (表达式 for 变量 in 序列)、tuple(生成器)

  参考答案：

  1
  # 生成器
  2
  gen = (x**2 for x in range(1, 11))
  3
  print(f"生成器对象: {gen}")
  4
  
  5
  # 转换为元组
  6
  t = tuple(gen)
  7
  print(f"元组: {t}")