Python 3.8+ 海象运算符详解
Python 3.8 引入了一个备受期待的新特性——海象运算符（Walrus Operator），语法为 :=。它允许你在表达式内部直接给变量赋值，并且返回这个值。这不仅仅是语法糖，更是为了减少重复计算和简化嵌套逻辑而设计的利器。
1 基础概念
海象运算符的核心在于'边算边存'。在传统写法中，我们往往需要先赋值再判断，或者在判断前调用一次函数，而在海象运算符出现后，这两步可以合并。它的本质是赋值表达式，既完成了赋值动作，又产出了结果供后续使用。
2 基础用法
2.1 条件语句中的赋值
以前写 if 判断时，如果判断条件涉及复杂的函数调用，我们不得不先把它赋给临时变量。现在可以直接写在 if 里。
# 传统写法
data = fetch_data()
if data:
process(data)
# 海象运算符写法
if (data := fetch_data()):
process(data)
注意括号的使用，虽然优先级上 := 较低，但加上括号能让意图更清晰，也能避免一些潜在的歧义。
2.2 循环中的赋值
在处理流式数据或需要检查状态的场景下，海象运算符能极大简化 while 循环。
# 传统写法
line = file.readline()
while line:
process(line)
line = file.readline()
# 海象运算符写法
while (line := file.readline()):
process(line)
这样不仅减少了重复代码，还让循环的终止条件一目了然。
2.3 表达式内部复用
有时候我们需要在一个复杂的表达式中多次用到同一个子表达式的结果。
# 传统写法
result = calculate(x)
if result > 0 and result < 100:
print(result)
# 海象运算符写法
if (result := calculate(x)) > 0 and result < 100:
print(result)
3 进阶技巧
3.1 列表推导式中的过滤
在列表推导式中，我们可以利用它来过滤掉无效数据，同时保留处理后的值。
valid_items = [x for x in items if (processed := transform(x)) is not None]
这里避免了两次调用 transform，也省去了中间变量的定义。
3.2 避免重复调用函数
对于昂贵的 API 调用或数据库查询，海象运算符能保证只执行一次。
if (user := get_user(id)) and user.is_active:
send_notification(user)
3.3 正则表达式匹配复用
正则匹配经常需要获取匹配对象以便提取组信息。
import re
pattern = re.compile(r'\d+')
text = "Price: 100"
if (match := pattern.search(text)):
print(match.group())
3.4 链式条件检查
可以将多个条件串联起来，保持逻辑紧凑。
if (val := get_val()) and val > 0 and validate(val):
save(val)
3.5 Walrus + f-string 调试
在调试阶段，可以直接在 f-string 中打印中间变量，无需额外打印语句。
print(f"Result: {res := compute()}")
4 注意事项
4.1 作用域规则
海象运算符创建的变量遵循 LEGB 规则。如果在函数外部使用，它会创建全局变量；在函数内部则是局部变量。但在列表推导式等特定作用域中，行为可能有所不同，需谨慎测试。
4.2 优先级陷阱
:= 的优先级非常低，甚至低于比较运算符。这意味着 x := y = 1 会报错，而 x := (y = 1) 才是合法的赋值链（尽管不推荐）。在复杂表达式中，务必使用括号明确意图。
4.3 不可用于顶级表达式
你不能在脚本的最顶层直接使用 := 进行赋值而不包含在语句中。例如 (x := 1) 在顶层是合法的，但单独一行 x := 1 在某些上下文中可能不被支持，通常建议包裹在语句块中。
4.4 设计哲学
PEP 572 的设计初衷是解决'赋值表达式'的需求，而不是鼓励滥用。如果代码变得难以阅读，请回归传统写法。可读性永远是第一位的。
5 总结
海象运算符是 Python 现代化进程中的重要一步。它让代码更紧凑，减少了样板代码。但记住，工具是为了服务逻辑，而非炫技。理解其背后的作用域和优先级机制，才能安全高效地使用它。
