Python:三元表达式、列表推导式和生成器表达式
三元表达式
语法格式
如下:
为真时的结果 if 判断条件 else 为假时的结果
例子
name = input('姓名>>: ')
res = '请进' if name == '张三' else '再见'
print(res)
结果如下
姓名>>: 张三
请进
姓名>>: 李四
再见
列表推导式
列表推导式用于使用其他列表创建一个新列表。
语法格式
如下:
# 格式1
[表达式 for 变量 in 列表]
# 格式2,对列表进行判断帅选
[表达式 for 变量 in 列表 if 条件]
例1
# 求1-9的平方组成的列表
l = [x**2 for x in range(1, 10)]
print(l)
结果如下:
[1, 4, 9, 16, 25, 36, 49, 64, 81]
例2
# 求1-9中偶数的平方组成的列表
l = [x**2 for x in range(1, 10) if x % 2 == 0]
print(l)
结果如下:
[4, 16, 36, 64]
生成器表达式
把列表推导式的[]
换成()
就是生成器表达式,返回对象不是一个列表,而是一个生成器。
优点:省内存,一次只产生一个值在内存中。
语法格式
如下:
# 格式1
(表达式 for 变量 in 列表)
# 格式2,对列表进行判断筛选
(表达式 for 变量 in 列表 if 条件)
例子
>>> l = (x**2 for x in range(1, 10) if x % 2 == 0)
>>> next(l)
4
>>> next(l)
16
>>> next(l)
36
>>> next(l)
64
>>> next(l)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
StopIteration