6 個例子教你重構(gòu) Python 代碼
劇照 | 《棋魂》
編寫干凈的 Pythonic 代碼就是盡可能使其易于理解，但又簡潔。所以重構(gòu)代碼讓代碼簡潔易理解是很有必要的。
本文重點(diǎn)是想跟大家探討為什么這些重構(gòu)是好想法，而不僅僅是如何做。
1. 合并嵌套的 if 條件
太多的嵌套會使代碼難以理解，這在 Python 中尤為如此，因?yàn)?Python 沒有括號來幫助區(qū)隔不同的嵌套級別。
閱讀深度嵌套的代碼容易讓人煩躁，因?yàn)槟惚仨毨砬迥男l件屬于哪一級。因此，我們應(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ù)的代碼行，并明確了條件實(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)常忽略的一個小竅門是 Python 的yield關(guān)鍵字有對應(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)
常見的模式是，我們需要查找是否集合中的一個或多個項(xiàng)符合某些條件。這可以通過 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)至少有一個元素計(jì)算為 True 時，any()?將返回 True，只有當(dāng)所有元素都計(jì)算為 True 時，all() 將返回 True。
如果對 any() 的調(diào)用找到一個值為 True 的元素，它可以立即返回。
5. 用[]替換list()
創(chuàng)建列表的最簡潔和 Pythonic 的方法是使用 []。
x?=?[]x = ['first', 'second']
這樣做有額外的優(yōu)點(diǎn)：是一個很好的改進(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)
這也是提高了程序性能 ，因?yàn)槿绻h(huán)中的任何語句將在每次循環(huán)運(yùn)行時執(zhí)行。在這些多次執(zhí)行上花費(fèi)的時間屬于浪費(fèi)，因?yàn)樗恍枰獔?zhí)行一次。如果語句涉及對數(shù)據(jù)庫的調(diào)用或其他耗時的任務(wù)，則這種節(jié)省可能非常重要。
參考鏈接：
https://sourcery.ai/blog/explaining-refactorings-1
近期熱門文章推薦：
