Python - PEP572: 海象運算符

海象運算符

• PEP572 的標題是「Assignment Expressions」，也就是「賦值表達式」，也叫做「命名錶達式」
• 不過它現在被廣泛的別名是「海象運算符」(The Walrus Operator)，因爲:=很像海象

賦值中間變量

未使用海象運算符

pattern = re.compile('s')
data = 'ss'
match = pattern.match(data)
if match:
    print(match.group(0))

使用海象運算符優化代碼

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)

文件讀取的例子

未使用海象運算符

while 1:
    line = fp.readline()
    if not line:
        break
    print(line)

使用海象運算符優化代碼

while (line := fp.readline()):
    print(line)

仍然做了四件事：

1. 對錶達式fp.readline()求值
2. 求值結果賦值給 line
3. 對 line 進行 while 循環的條件判斷，如果 None 就退出循環
4. line 變量可以在 while 循環體內使用

這個就真的簡化了很多的代碼

簡化列表解析

常見使用添加數據到列表的寫法

results = []
for x in data:
    result = f(x)
    if result:
        results.append(result)

假設想使用列表生成式來解決呢？

results = [
    f(x) for x in data
    if f(x)
]

這樣是錯誤的，爲什麼？因爲會執行兩次f(x)，不符合實際需求
可以使用海象表達式來優化

results = [
    y for x in data 
    if (y := f(x))
]

官方例子

stuff = [[y := f(x), x/y] for x in range(5)]

因爲每一項包含了 y，以及要用 y 才能獲得結果 x/y，所以 賦值給中間變量是海象運算符的關鍵

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