前言

對(duì)自己寫(xiě)的冗長(zhǎng)代碼，想重構(gòu)但又無(wú)思路？小編整理了介紹python代碼重構(gòu)優(yōu)化的一些方法，助你一臂之力。

編寫(xiě)干凈的 Pythonic 代碼就是盡可能使其易于理解，但又簡(jiǎn)潔。所以重構(gòu)代碼讓代碼簡(jiǎn)潔易理解是很有必要的。

本文重點(diǎn)是想跟大家探討為什么這些重構(gòu)是好想法，而不僅僅是如何做。

1. 合并嵌套的 if 條件

太多的嵌套會(huì)使代碼難以理解，這在 Python 中尤為如此，因?yàn)?Python 沒(méi)有括號(hào)來(lái)幫助區(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ǔ)句之外

我們應(yīng)該始終尋找移除重復(fù)代碼的方法。這是提升代碼能力的好辦法。

有時(shí)，在條件的兩個(gè)分支上出現(xiàn)重復(fù)代碼，這意味這段代碼將始終執(zhí)行。因此這段重復(fù)的代碼可以從條件中移出，放在條件之外執(zhí)行。

if sold > DISCOUNT_AMOUNT:
 total = sold * DISCOUNT_PRICE
 label = f'Total: {total}'
else:
 total = sold * PRICE
 label = f'Total: {total}'

通過(guò)將對(duì)label變量賦值移到到條件之外，我們刪除了重復(fù)的代碼行，并明確了條件實(shí)際控制的內(nèi)容，就是計(jì)算total。

if sold > DISCOUNT_AMOUNT:
  total = sold * DISCOUNT_PRICE
else:
  total = sold * PRICE
label = f'Total: {total}'

3. 將內(nèi)部循環(huán)中的yield替換為yield from

經(jīng)常忽略的一個(gè)小竅門(mén)是 Python 的yield關(guān)鍵字有對(duì)應(yīng)的為collections準(zhǔn)備的yield from。因此無(wú)需使用 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)

常見(jiàn)的模式是，我們需要查找是否集合中的一個(gè)或多個(gè)項(xiàng)符合某些條件。這可以通過(guò) for 循環(huán)完成，例如：

found = False
for thing in things:
  if thing == other_thing:
    found = True
    break

更簡(jiǎn)潔的方法，是使用 Python 的 any() 和 all()內(nèi)置函數(shù)，來(lái)清楚地顯示代碼的意圖。

found = any(thing == other_thing for thing in things)

當(dāng)至少有一個(gè)元素計(jì)算為 True 時(shí)，all() 將返回 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í)行的語(yǔ)句移出for/while循環(huán)

將“不變的”語(yǔ)句從循環(huán)中移出。如果某條語(yǔ)句只是設(shè)置了一些變量供循環(huán)使用，則不需要在循環(huán)中。循環(huán)本身就是復(fù)雜的，因此在編寫(xiě)循環(huán)時(shí)，應(yīng)牢記，使其更短、更容易理解。

在此示例中，city變量在循環(huán)中賦值，但它只讀取且不更改。

for building in buildings:
  city = 'London'
  addresses.append(building.street_address, city)

因此，將其移出是安全的，這更清楚地說(shuō)明，相同的city值將應(yīng)用于每個(gè)building變量。

city = 'London'
for building in buildings:
  addresses.append(building.street_address, city)

這也是提高了程序性能 ，因?yàn)槿绻h(huán)中的任何語(yǔ)句將在每次循環(huán)運(yùn)行時(shí)執(zhí)行。在這些多次執(zhí)行上花費(fèi)的時(shí)間屬于浪費(fèi)，因?yàn)樗恍枰獔?zhí)行一次。如果語(yǔ)句涉及對(duì)數(shù)據(jù)庫(kù)的調(diào)用或其他耗時(shí)的任務(wù)，則這種節(jié)省可能非常重要。

參考鏈接：

https://sourcery.ai/blog/explaining-refactorings-1/

總結(jié)

到此這篇關(guān)于重構(gòu)Python代碼的文章就介紹到這了,更多相關(guān)重構(gòu)Python代碼內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論