6 個(gè)例子教你重構(gòu) Python 代碼
【導(dǎo)語】:對(duì)自己寫的冗長(zhǎng)代碼,想重構(gòu)但又無思路?小編整理了介紹python代碼重構(gòu)優(yōu)化的一些方法,助你一臂之力。
編寫干凈的 Pythonic 代碼就是盡可能使其易于理解,但又簡(jiǎn)潔。所以重構(gòu)代碼讓代碼簡(jiǎn)潔易理解是很有必要的。
本文重點(diǎn)是想跟大家探討為什么這些重構(gòu)是好想法,而不僅僅是如何做。
1. 合并嵌套的 if 條件
太多的嵌套會(huì)使代碼難以理解,這在 Python 中尤為如此,因?yàn)?Python 沒有括號(hào)來幫助區(qū)隔不同的嵌套級(jí)別。
閱讀深度嵌套的代碼容易讓人煩躁,因?yàn)槟惚仨毨砬迥男l件屬于哪一級(jí)。因此,我們應(yīng)盡可能減少嵌套,如果兩個(gè)條件可以用 and 合在一起,會(huì)比嵌套更易讀。
合并之前:
if a:if b:return c
合并后:
if a and b:return c
2. 將重復(fù)的代碼移到條件語句之外
我們應(yīng)該始終尋找移除重復(fù)代碼的方法。這是提升代碼能力的好辦法。
有時(shí),在條件的兩個(gè)分支上出現(xiàn)重復(fù)代碼,這意味這段代碼將始終執(zhí)行。因此這段重復(fù)的代碼可以從條件中移出,放在條件之外執(zhí)行。
if sold > DISCOUNT_AMOUNT:total = sold * DISCOUNT_PRICElabel = f'Total: {total}'else:total = sold * PRICElabel = f'Total: {total}'
通過將對(duì)label變量賦值移到到條件之外,我們刪除了重復(fù)的代碼行,并明確了條件實(shí)際控制的內(nèi)容,就是計(jì)算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)常忽略的一個(gè)小竅門是 Python 的yield關(guān)鍵字有對(duì)應(yīng)的為collections準(zhǔn)備的yield from。因此無需使用 for 循環(huán)遍歷集合。這使代碼變短,并刪除 for 中的額外變量。而且消除 for 循環(huán)后,yield from使程序運(yùn)行效率提高約 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)
常見的模式是,我們需要查找是否集合中的一個(gè)或多個(gè)項(xiàng)符合某些條件。這可以通過 for 循環(huán)完成,例如:
found = Falsefor thing in things:if thing == other_thing:found = Truebreak
更簡(jiǎn)潔的方法,是使用 Python 的 any() 和 all()內(nèi)置函數(shù),來清楚地顯示代碼的意圖。
found = any(thing == other_thing for thing in things)當(dāng)至少有一個(gè)元素計(jì)算為 True 時(shí),any()?將返回 True,只有當(dāng)所有元素都計(jì)算為 True 時(shí),all() 將返回 True。
如果對(duì) any() 的調(diào)用找到一個(gè)值為 True 的元素,它可以立即返回。
5. 用[]替換list()
創(chuàng)建列表的最簡(jiǎn)潔和 Pythonic 的方法是使用 []。
x?=?[]x = ['first', 'second']
這樣做有額外的優(yōu)點(diǎn):是一個(gè)很好的改進(jìn)程序性能的方法。
以下是更改之前和之后的時(shí)間對(duì)比:
$ 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)時(shí),應(yīng)牢記,使其更短、更容易理解。
在此示例中,city變量在循環(huán)中賦值,但它只讀取且不更改。
for building in buildings:city = 'London'addresses.append(building.street_address, city)
因此,將其移出是安全的,這更清楚地說明,相同的city值將應(yīng)用于每個(gè)building變量。
city = 'London'for building in buildings:addresses.append(building.street_address, city)
這也是提高了程序性能 ,因?yàn)槿绻h(huán)中的任何語句將在每次循環(huán)運(yùn)行時(shí)執(zhí)行。在這些多次執(zhí)行上花費(fèi)的時(shí)間屬于浪費(fèi),因?yàn)樗恍枰獔?zhí)行一次。如果語句涉及對(duì)數(shù)據(jù)庫的調(diào)用或其他耗時(shí)的任務(wù),則這種節(jié)省可能非常重要。
參考鏈接:
https://sourcery.ai/blog/explaining-refactorings-1/
關(guān)于代碼重構(gòu),你們又有哪些心得想法?歡迎在評(píng)論中和我探討。覺得文章不錯(cuò),請(qǐng)點(diǎn)贊和在看支持我繼續(xù)分享好文。謝謝!
PS:公號(hào)內(nèi)回復(fù)「Python」即可進(jìn)入Python 新手學(xué)習(xí)交流群,一起?100 天計(jì)劃!
老規(guī)矩,兄弟們還記得么,右下角的 “在看” 點(diǎn)一下,如果感覺文章內(nèi)容不錯(cuò)的話,記得分享朋友圈讓更多的人知道!


【神秘禮包獲取方式】
