優(yōu)化重構(gòu)Python代碼6個小建議
大家好,歡迎來到 Crossin的編程教室 !
編寫干凈的 Pythonic 代碼就是盡可能使其既簡潔又易于理解。所以當(dāng)你學(xué)習(xí)編程到一定階段之后,重構(gòu)代碼是很有必要的。
但不少經(jīng)驗尚不豐富的學(xué)習(xí)者對于自己寫的冗長代碼,想重構(gòu)但又無思路。今天我們就來分享幾個python代碼重構(gòu)優(yōu)化的小技巧,助你一臂之力。
本文重點是想跟大家探討為什么這些重構(gòu)是好想法,而不僅僅是如何做。
1. 合并嵌套的 if 條件
太多的嵌套會使代碼難以理解,這在 Python 中尤為如此,因為 Python 沒有括號來幫助區(qū)隔不同的嵌套級別。
閱讀深度嵌套的代碼容易讓人煩躁,因為你必須理清哪些條件屬于哪一級。因此,我們應(yīng)盡可能減少嵌套,如果兩個條件可以用 and 合在一起,會比嵌套更易讀。
合并之前:
if a:if b:return c
合并后:
if a and b:return c
2. 將重復(fù)的代碼移到條件語句之外
我們應(yīng)該始終尋找移除重復(fù)代碼的方法。這是提升代碼能力的好辦法。
有時,在條件的兩個分支上出現(xiàn)重復(fù)代碼,這意味這段代碼將始終執(zhí)行。因此這段重復(fù)的代碼可以從條件中移出,放在條件之外執(zhí)行。
if sold > DISCOUNT_AMOUNT:total = sold * DISCOUNT_PRICElabel = f'Total: {total}'else:total = sold * PRICElabel = f'Total: {total}'
通過將對label變量賦值移到到條件之外,我們刪除了重復(fù)的代碼行,并明確了條件實際控制的內(nèi)容,就是計算total。
if sold > DISCOUNT_AMOUNT:total = sold * DISCOUNT_PRICEelse:total = sold * PRICElabel = f'Total: {total}'
3. 將內(nèi)部循環(huán)中的 yield 替換為 yield from
經(jīng)常忽略的一個小竅門是 Python 的yield關(guān)鍵字有對應(yīng)的為collections準(zhǔn)備的yield from。因此無需使用 for 循環(huán)遍歷集合。這使代碼變短,并刪除 for 中的額外變量。而且消除 for 循環(huán)后,yield from使程序運行效率提高約 15%。
重構(gòu)前:
def get_content(entry):for block in entry.get_blocks():yield block
重構(gòu)后:
def get_content(entry):yield from entry.get_blocks()
4. 使用 any() 而不是用于循環(huán)
常見的模式是,我們需要查找是否集合中的一個或多個項符合某些條件。這可以通過 for 循環(huán)完成,例如:
found = Falsefor thing in things:if thing == other_thing:found = Truebreak
更簡潔的方法,是使用 Python 的 any() 和 all() 內(nèi)置函數(shù),來清楚地顯示代碼的意圖。
found = any(thing == other_thing for thing in things)當(dāng)至少有一個元素計算為 True 時,any() 將返回 True,只有當(dāng)所有元素都計算為 True 時,all() 將返回 True。
如果對 any() 的調(diào)用找到一個值為 True 的元素,它可以立即返回。
5. 用 [ ] 替換 list()
創(chuàng)建列表的最簡潔和 Pythonic 的方法是使用 []。
x = []x = ['first', 'second']
這樣做有額外的優(yōu)點:是一個很好的改進(jìn)程序性能的方法。
以下是更改之前和之后的時間對比:
$ python3 -m timeit "x = list()"5000000 loops, best of 5: 63.3 nsec per loop
$ python3 -m timeit "x = []"20000000 loops, best of 5: 15.8 nsec per loop
同樣的原因和性能表現(xiàn),使用 { } 替代 dict()。
6. 將重復(fù)執(zhí)行的語句移出 for / while 循環(huán)
將“不變的”語句從循環(huán)中移出。如果某條語句只是設(shè)置了一些變量供循環(huán)使用,則不需要在循環(huán)中。循環(huán)本身就是復(fù)雜的,因此在編寫循環(huán)時,應(yīng)牢記,使其更短、更容易理解。
在此示例中,city變量在循環(huán)中賦值,但它只讀取且不更改。
for building in buildings:city = 'London'addresses.append(building.street_address, city)
因此,將其移出是安全的,這更清楚地說明,相同的city值將應(yīng)用于每個building變量。
city = 'London'for building in buildings:addresses.append(building.street_address, city)
這也是提高了程序性能 ,因為如果循環(huán)中的任何語句將在每次循環(huán)運行時執(zhí)行。在這些多次執(zhí)行上花費的時間屬于浪費,因為它只需要執(zhí)行一次。如果語句涉及對數(shù)據(jù)庫的調(diào)用或其他耗時的任務(wù),則這種節(jié)省可能非常重要。
關(guān)于代碼重構(gòu),你們又有哪些心得想法?歡迎在評論一起探討。如果文章對你有幫助,歡迎轉(zhuǎn)發(fā)/點贊/收藏~
參考鏈接:
https://sourcery.ai/blog/explaining-refactorings-1/
作者:Python開發(fā)者
_往期文章推薦_
