Python表达式、生成式、推导式简介

文章目录

• 一、三元表达式
• • 1. 语法
  • 2. 实例
• 二、列表生成式（又称列表推导式）
• • 1. 语法
  • 2. 作用
  • 3. 列表推导式实例
  • 4. 集合推导式
  • 5. 字典推导式
• 三、生成器表达式
• • 1. 语法
  • 2. 实例
• 结语

一、三元表达式

1. 语法

在C和Java中的三元运算符语法如下：

判段的条件 ? 条件为真时的结果:条件为假时的结果

Python中没有像C和Java等语言中的三元运算符。

但是Python中有三元表达式来简化代码，其语法如下：

res = 条件成立时返回的值 if 条件 else 条件不成立时返回的值

2. 实例

针对以下场景

def max1(x,y):
    if x > y:
        return x
    else:
        return y

res = max1(1,2)

用三元表达式可以一行解决。

x=1
y=2
res = x if x > y else y # 三元表达式

二、列表生成式（又称列表推导式）

List Comprehension ，列表生成式（又称列表推导式、列表解析式）提供了一种简明扼要的方法来创建列表。

虽然名字叫做 List Comprehension，但是这个语法同样适用于dict、set等一系列可迭代的数据结构。

1. 语法

列表生成式是python为我们提供的一种简化代码的解决方案，用来快速生成列表，语法如下：

[表达式 for 迭代变量 in 可迭代对象 [if 条件表达式] ]

此格式中，[if 条件表达式] 不是必须的，可以使用，也可以省略。

除去 [if 条件表达式] 部分，其余各部分的含义以及执行顺序和 for 循环是完全一样的（表达式其实就是 for 循环中的循环体），即它的执行顺序如下所示：

for 迭代变量 in 可迭代对象
    表达式

列表生成式与 for 循环语句最大的不同之处在于，列表推导式最终会将循环过程中，计算表达式得到的一系列值组成一个列表。

2. 作用

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

对一个列表（集合，字典等）根据条件进行循环遍历，把符合条件的筛选出来并生成新的列表（集合，字典等）。

括号内可以写for循环和if-else并且可以多层嵌套for、if，甚至可以在里面再嵌套一个推导式。

3. 列表推导式实例

针对以下场景：

egg_list=[]
for i in range(10):
    egg_list.append('鸡蛋%s' %i)

用列表生成式可以一行解决

egg_list=['鸡蛋%s' %i for i in range(10)]

print(egg_list)的结果：

4. 集合推导式

集合推导式使用的是花括号，用法和列表推导式一样：

str1 = ['aaaa', 'bbb', 'cccc', 'ddd', 'eee', 'ffff', 'ggggg']
formula = {str2 for str2 in str1 if len(str2) >= 4}
print(formula)

结果：{‘ffff’, ‘cccc’, ‘aaaa’, ‘ggggg’}或{‘cccc’, ‘aaaa’, ‘ggggg’, ‘ffff’}等等。

注意：集合是一个无序不重复元素集。

5. 字典推导式

字典推导式和集合推导式一样使用花括号，区别是字典推导式同时取出键和值，中间用冒号分开。

还要注意字典要使用items()函数，该函数将字典的键和值以列表方式返回。

dict1 = {
    'name': 'Tom',
    'age': 18,
    'sex': 'man',
}
formula = {key: value for key, value in dict1.items() if key == 'name'}
print(formula)

运行结果：{‘name’: ‘Tom’}

三、生成器表达式

创建一个生成器对象有两种方式，一种是调用带yield关键字的函数，另一种就是生成器表达式。

生成器简介

1. 语法

与列表生成式的语法格式相同，只需要将[ ]换成( )，即：

（expression for item in iterable if condition）

对比列表生成式返回的是一个列表，生成器表达式返回的是一个生成器对象。

>>> [x*x for x in range(3)]
[0, 1, 4]
>>> g=(x*x for x in range(3))
>>> g
<generator object <genexpr> at 0x101be0ba0>

2. 实例

如果我们要读取一个大文件的字节数，应该基于生成器表达式的方式完成：

with open('db.txt','rb') as f:
    nums=(len(line) for line in f)
    total_size=sum(nums) # 依次执行next(nums)，然后累加到一起得到结果