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

• 2019 年 11 月 28 日
• 筆記

作　者 | Jonathan Hsu

翻　譯 | 人氣呆毛選手

審　校 | 鳶尾、唐里

註：敬請點擊文末【閱讀原文】訪問文中相關鏈接，PC查看體驗更佳。

運算符使值的賦值可以傳遞到表達式中。這通常會使語句數減少一個。例如：

  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)

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

總結

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

via https://medium.com/better-programming/three-ways-to-use-the-walrus-operator-in-python-d5550f3a7dd