Python 小技巧 039：使用 “赋值表达式”

zltzlt

Python 的赋值表达式

“赋值表达式” 是 Python 3.8 新增的语法，今天我们来学习它的用法。

1. 语法

这个运算符长这样：

:=

:= 可以在表达式内部为变量赋值。因为它很像海象的眼睛和长牙，所以它被亲切地称为 “海象运算符”。

语法格式：

(a := b)    # 括号不能少

其中，a 为合法的标识符。b 为一个常量或变量。这个表达式的值为 b，而且能将 b 的值赋给 a。

2. 实际运用

让我们先来看看它怎么使用：

>>> # 前面没有对 a 赋值
>>> b = 'FishC'
>>> (a := b)    # 相当于 a = b; b
'FishC'
>>> a
'FishC'
>>> c = 'I love FishC.com'
>>> a := c    # 少了对括号，报错
SyntaxError: invalid syntax

:= 的运用非常广泛。举个例子，在 if 分支中：

a = [1, 2, 3, 4, 5]
if (n := len(a)) > 3:
    print(f"列表 a 过长，它的长度为 {n}")

# 运行结果：列表 a 过长，它的长度为 5

在这个例子中，赋值表达式可以避免调用两次 len()。

正则表达式匹配中，在需要使用两次匹配对象 Match 的情况下（一次用于检测是否匹配，另一次用于提取匹配的字符串）也可以用 :=：

import re
if (match := re.search(r"\d+", "abcd123")) is not None:
    print(match.group())

# 运行结果：123

:= 在 while 循环中的应用：

f = open("file.txt", encoding="utf-8")
while (block := f.readline()) != "":
    print(block[:-1])

在列表推导式中也可以用到 :=：

names, allowed_names = ["lis", "jckson", "python", "canad"], ["lisa", "jackson", "canada"]
normalize = lambda x: x + "a"
list1 = [name2.title() for name in names if (name2 := normalize(name)) in allowed_names]
print(" ".join(list1))

# 运行结果：Lisa Canada

运用 := 可以降低代码的复杂性，提升代码的可读性。

hrp

原来还有这个用法

freeck2013

赞 re那个 每次都重新定义match 好烦 这样好多了