不少人不知道的Python 炫技操做：海象運算符的三種用法

首發於公衆號：《Python編程時光》，一個專一輸出高質量 Python開發知識的專欄

系列導讀

1. 不少人不知道的Python 炫技操做：條件語句的七種寫法
2. 不少人不知道的Python 炫技操做：合併字典的七種方法

---

Python 版本發展很是快，現在最新的版本已是 Pyhton 3.9，即使如此，有不少人甚至還停留在 3.6 或者 3.7，連 3.8 還沒用上。

不少 Python 3.8 的特性還沒來得及瞭解，就已經成爲舊知識了，好比今天要說的海象運算符。

海象運算符是在 PEP 572 被提出的，直到 3.8 版本合入發佈。

它的英文原名叫 Assignment Expressions，翻譯過來也就是 賦值表達式，不過如今你們更廣泛地稱之爲海象運算符，就是由於它長得真的太像海象了。

1. 第一個用法：if/else

可能有朋友是第一次接觸這個新特性，因此仍是簡單的介紹一下這個海象運算符有什麼用？

在 Golang 中的條件語句能夠直接在 if 中運算變量的獲取後直接對這個變量進行判斷，可讓你少寫一行代碼

import "fmt"

func main() {
    if age := 20;age > 18 {
        fmt.Println("已經成年了")
    }
}
複製代碼

若在 Python 3.8 以前，Python 必須得這樣子寫

age = 20
if age > 18:
    print("已經成年了")
複製代碼

但有了海象運算符以後，你能夠和 Golang 同樣（若是你沒學過 Golang，那這裏要注意，Golang 中的 := 叫短變量聲明，意思是聲明並初始化，它和 Python 中的 := 不是一個概念）

if (age:= 20) > 18:
    print("已經成年了")
複製代碼

2. 第二個用法：while

在不使用 海象運算符以前，使用 while 循環來讀取文件的時候，你也許會這麼寫

file = open("demo.txt", "r")
while True:
    line = file.readline()
    if not line:
        break
    print(line.strip())
複製代碼

但有了海象運算符以後，你能夠這樣

file = open("demo.txt", "r")
while (line := file.readline()):
    print(line.strip())
複製代碼

使用它替換以往的無限 while 循環寫法更爲驚豔

好比，實現一個須要命令行交互輸入密碼並檢驗的代碼，你也許會這樣子寫

while True:
   p = input("Enter the password: ")
   if p == "youpassword":
      break
複製代碼

有了海象運算符以後，這樣子寫更爲舒服

while (p := input("Enter the password: ")) != "youpassword":
   continue
複製代碼

3. 第三個用法：推導式

這個系列的文章，幾乎每篇都能看到推導式的身影，這一篇依舊如此。

在編碼過程當中，我很喜歡使用推導式，在簡單的應用場景下，它簡潔且不失高效。

以下這段代碼中，我會使用列表推導式得出全部會員中過於肥胖的人的 bmi 指數

members = [
    {"name": "小五", "age": 23, "height": 1.75, "weight": 72},
    {"name": "小李", "age": 17, "height": 1.72, "weight": 63},
    {"name": "小陳", "age": 20, "height": 1.78, "weight": 82},
]

count = 0

def get_bmi(info):
    global count
    count += 1

    print(f"執行了 {count} 次")

    height = info["height"]
    weight = info["weight"]

    return weight / (height**2)

# 查出全部會員中過於肥胖的人的 bmi 指數
fat_bmis = [get_bmi(m) for m in members if get_bmi(m) > 24]

print(fat_bmis)
複製代碼

輸出以下

執行了 1 次
執行了 2 次
執行了 3 次
執行了 4 次
[25.88057063502083]
複製代碼

能夠看到，會員數只有 3 個，可是 get_bmi 函數卻執行了 4 次，緣由是在判斷時執行了 3 次，而在構造新的列表時又重複執行了一遍。

若是全部會員都是過於肥胖的，那最終將執行 6 次，這種在大量的數據下是比較浪費性能的，所以對於這種結構，我一般會使用傳統的for 循環 + if 判斷。

fat_bmis = []

# 查出全部會員中過於肥胖的人的 bmi 指數
for m in members:
    bmi = get_bmi(m)
    if bmi > 24:
        fat_bmis.append(bmi)
複製代碼

在有了海象運算符以後，你就能夠不用在這種場景下作出妥協。

# 查出全部會員中過於肥胖的人的 bmi 指數
fat_bmis = [bmi for m in members if (bmi := get_bmi(m)) > 24]
複製代碼

最終從輸出結果能夠看出，只執行了 3 次

執行了 1 次
執行了 2 次
執行了 3 次
[25.88057063502083]
複製代碼

這裏僅介紹了列表推導式，但在字典推導式和集合推導式中一樣適用。再也不演示。

海象運算符，是一個新奇的特性，有很多人以爲這樣這種特性會破壞代碼的可讀性。確實在一個新鮮事物剛出來時是會這樣，但我相信通過時間的沉澱後，愈來愈多的人使用它並享受它帶來的便利時，這種爭議也會慢慢消失在歷史的長河中。

---