Python 怎么用海象运算符(:=)简化循环中的重复计算？(3.8+)

Question

仓乃文 · Accepted Answer

循环中先计算再判断时，需要先赋值给变量再使用，写两遍同样的表达式。海象运算符 := 在表达式内部同时完成赋值和返回值，消除重复计算，让 while 和列表推导更紧凑。

简化 while 循环：
```python
# 之前：重复调用
line = input()
while line != "quit":
    process(line)
    line = input()

# 海象运算符：一次搞定
while (line := input()) != "quit":
    process(line)
```

列表推导中避免重复计算：
```python
# 之前：expensive_cleanup 调用两次
results = [expensive_cleanup(x) for x in data if expensive_cleanup(x) is not None]

# 海象运算符：只调用一次
results = [cleaned for x in data if (cleaned := expensive_cleanup(x)) is not None]
```

读取数据直到空值：
```python
chunks = []
while (chunk := read_chunk()):
    chunks.append(chunk)
```

补充说明：
- 海象运算符是 Python 3.8 引入的，:= 形似海象眼睛和象牙
- 核心作用：先计算、再赋值、再判断，一步到位
- 不要嵌套使用，会严重降低可读性
- 适合 while 循环条件、列表推导过滤、读取流式数据
- if 语句中也可以用：if (n := len(data)) > 10: print(n)