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