Python 海象運算子 (:=) 的三種用法

語言: CN / TW / HK

隨著Python 3.8的釋出,賦值表示式運算子(也稱為海象運算子)也釋出了。
運算子使值的賦值可以傳遞到表示式中。這通常會使語句數減少一個。例如:

my_list = [1,2,3]
count = len(my_list)
if count > 3:
   print(f"Error, {count} is too many items")

# when converting to walrus operator...
if (count := len(my_list)) > 3:
   print(f"Error, {count} is too many items")

如果您快速瀏覽了一下,就會理解對這個新操作的擔憂。似乎將兩個簡單的語句混為一談,以使收益最小。
假設對於為什麼引入此運算子有合理答案?那麼讓人很自然的想到,它在現實世界中如何有用?
我發現了我將要繼續採用海象運算子的三個方面,且每個方面除了能省一行程式碼之外都還能帶來其他好處。

While 迴圈
雖然迴圈似乎是最常見的使用海象運算子的情況。在此處使用海象運算子的好處是可以在while語句中合併表示式和修飾符。我已經寫過無數次無限迴圈了,因為我總是忘記在迴圈末尾寫修改語句。

line = f.readLine()
while line:
   print(line)
   line = f.readLine()

# when converting to walrus operator...while line := f.readLine():
   print(line)

不過要注意, 使用海象運算子更適合用於傳統的do/while 迴圈,因為賦值在迴圈表示式之前。請參見下面的差異:

n = 0
while n < 3:
   print(n) # 0,1,2
   n += 1

# when converting to walrus operator...
w = 0
while (w := w + 1) < 3:
   print(w) # 1,2

根據我有限的經驗,我發現它在替換無限while迴圈中最有用:

while True:
   p = input("Enter the password: ")
   if p == "the password":
      break

# when converting to walrus operator...
while (p := input("Enter the password: ")) != "the password":
   continue

能夠將while迴圈轉換為一個語句實在太棒啦。

列表理解
我發現海象運算子對優化某些列表理解語句很有幫助。有兩個標準1)需要過濾資料,2)儲存taxing function的結果。讓我們看一下傳統的列表理解語句:

scores = [22,54,75,89]
valid_scores = [
   longFunction(n)
   for n in scores
   if longFunction(n)
]

注意條件語句longFunction(n)了嗎?注意longFunction()被呼叫了兩次嗎?這是海象運算子的候選。

scores = [22,54,75,89]
valid_scores = [
   result for n in scores
   result := longFunction(n)
]

在優化了的程式碼中,longFunction()僅被呼叫一次,隱含的降低了呼叫次數。

處理返回的資料
接下來是我認為海象運算子的最有用的實現。我經常使用稱為iFormBuilder的移動資料收集軟體。具體來說,請求記錄(資料行)後,查詢返回資料時,需要對其進行處理。這是我每天編寫的程式碼塊:

# look for failed inspections
# if there are failed inspections, assign to technicianrecords = api.readFailedRecords()
if len(records) > 0:
   for record in records:
      api.assignToTechnician(record)

我仔細閱讀記錄,認真謹慎的敲程式碼,以確保僅當返回列表有資料該程式碼塊才執行:

if records := api.readFailedRecords():
   for record in records:
      api.assignToTechnician(record)

就可讀性而言,此重構的破壞性最小,實際上,我發現在這種情況下,海象運算子更易於閱讀。與前兩個示例形成鮮明對比的是,前兩個示例對可讀性的影響較小,有利於省心和提高效能。

總結
海象運算子很新穎並且有爭議,但是隻有時間會見證它逐漸被大家接收。你是否已開始在程式碼中使用它了呢?

以上就是本次分享的所有內容,想要了解更多 python 知識歡迎前往公眾號:Python 程式設計學習圈 ,傳送 “J” 即可免費獲取,每日干貨分享