Python - PEP572: 海象运算符

海象运算符

  • PEP572 的标题是「Assignment Expressions」,也就是「赋值表达式」,也叫做「命名表达式」
  • 不过它现在被广泛的别名是「海象运算符」(The Walrus Operator),因为:=很像海象

赋值中间变量

未使用海象运算符

python 复制代码
pattern = re.compile('s')
data = 'ss'
match = pattern.match(data)
if match:
    print(match.group(0))

使用海象运算符优化代码

python 复制代码
pattern = re.compile('s')
data = 'ss'
if match := pattern.match(data):
    print(match.group(0))

使用海象运算符之后,分别做了四件事

  1. 对表达式pattern.match(data)求值
  2. 把第一步的求值结果赋值给 match 变量
  3. 把 match 变量作为 if 的 条件,判断它的值是不是 None
  4. 而 match 变量可以在 if 的作用域范围内继续使用 print(match.group(0)

文件读取的例子

未使用海象运算符

python 复制代码
while 1:
    line = fp.readline()
    if not line:
        break
    print(line)

使用海象运算符优化代码

python 复制代码
while (line := fp.readline()):
    print(line)

仍然做了四件事:

  1. 对表达式fp.readline()求值
  2. 求值结果赋值给 line
  3. 对 line 进行 while 循环的条件判断,如果 None 就退出循环
  4. line 变量可以在 while 循环体内使用

这个就真的简化了很多的代码

简化列表解析

常见使用添加数据到列表的写法

python 复制代码
results = []
for x in data:
    result = f(x)
    if result:
        results.append(result)

假设想使用列表生成式来解决呢?

python 复制代码
results = [
    f(x) for x in data
    if f(x)
]

这样是错误的,为什么?因为会执行两次f(x),不符合实际需求

可以使用海象表达式来优化

python 复制代码
results = [
    y for x in data 
    if (y := f(x))
]

官方例子

python 复制代码
stuff = [[y := f(x), x/y] for x in range(5)]

因为每一项包含了 y,以及要用 y 才能获得结果 x/y,所以 赋值给中间变量是海象运算符的关键

一段迷惑的使用海象操作符的代码

相关推荐
cooldream20092 小时前
Python 包管理新时代:深入了解 `uv` 的使用与实践
python·uv·包管理器
之歆2 小时前
Python-魔术方法-创建、初始化与销毁-hash-bool-可视化-运算符重载-容器和大小-可调用对象-上下文管理-反射-描述器-二分-学习笔记
笔记·python·学习
胖达不服输2 小时前
「日拱一码」025 机器学习——评价指标
人工智能·python·机器学习·评价指标
brave_zhao3 小时前
JavaBeanUtils javaBean转map, 实体类转map,实体集合转List<Map>
linux·windows·python
apihz4 小时前
通用图片搜索-搜狗源免费API接口使用指南
android·java·python·php·音视频
爱吃面条的猿4 小时前
pycharm中自动补全方法返回变量
ide·python·pycharm
倔强青铜三5 小时前
苦练Python第15天:Lambda函数——Python的匿名一行杀器
人工智能·python·面试
云空5 小时前
《QtPy:Python与Qt的完美桥梁》
开发语言·python·qt·pyqt
Hi_kenyon5 小时前
UV的使用总结
python·uv
暴走的海鸽5 小时前
存储库模式赋能 Django:让你的代码不那么业余,更具生命力
python·设计模式·django