一日一技:巧用or關鍵字實現多重條件判斷
在Python開發過程中,我們難免會遇到多重條件判斷的情況的情況,此時除了用很多的if...elif...else以外,還可以巧用or關鍵字實現。
例如,我們要獲取天氣信息,現在有三個api接口。
- api_1()
- api_2()
- api_3()
其中api_1()的數據又詳細又好,但是接口不穩定,可能隨時會掉線。
api_2()返回的結果比api_1()稍微差一點,但是穩定性還不錯。
api_3()接口返回的是兜底數據,就是在前兩個接口都故障的情況下才會使用。
如果前兩個接口正常工作時返回字典,故障時他們都會返回None。
現在我們要獲取天氣信息,你可能會這樣寫代碼:
- weather = api_1()
- if not weather:
- weather = api_2()
- if not weather:
- weather = api_3()
代碼這樣寫,功能自然是沒有問題,但是非常難看。
實際上,代碼可以寫為:
- weather = api_1() or api_2() or api_3()
只需要1行代碼就能實現自動選擇。如果api_1()返回的是一個非空字典,那么由于短路效應,后面的兩個函數都不會運行。當api_1返回None時,api_2()才會運行。并且如果api_2()返回非空字典,那么api_3()依然不會運行。
我們現在在Jupyter里面測試一下,首先看看如果api_1()直接返回非空字典的情況,如下圖所示。
可以看到,在api_2()和api_3里面的兩行print語句都沒有執行。
接下來我們看看api_1()故障時候的情況。
此時先運行api_1()返回None,再運行api_2()返回非空字典。符合預期。
當前兩個函數都返回None的時候,第三個函數才會運行,如下圖所示。
除了做多重條件判斷外,還可以用來自己和自己取或操作,實現重試。
例如api_1()可能成功也可能失敗,所以需要嘗試運行3次,那么代碼可以這樣寫:
- weather = api_1() or api_1() or api_1()
本文轉載自微信公眾號「未聞Code」,可以通過以下二維碼關注。轉載本文請聯系未聞Code公眾號。