【導語】：對自己寫的冗長代碼，想重構(gòu)但又無思路？小編整理了系列介紹python代碼重構(gòu)優(yōu)化的方法，助你一臂之力。

編寫干凈的 Pythonic 代碼就是盡可能使其易于理解，但又簡潔。

這是 Python 重構(gòu)系列中的之一，本系列文章的重點是為什么這些重構(gòu)是好主意，而不僅僅是如何做。

1.將for循環(huán)轉(zhuǎn)換為list/dictionary/set 表達式

我們在時經(jīng)常遇到的一個情況是，創(chuàng)建一個值的集合。

比如我們創(chuàng)建一個列表，然后迭代地用值填充它，這里我們想創(chuàng)建一個立方數(shù)字的列表。大多數(shù)語言的標準方法如下：

cubes = []
for i in range(20):
    cubes.append(i ** 3)

在Python中，我們可以使用列表表達式，生成需要的數(shù)據(jù)。就可以將代碼簡化為一行，省去定義列表，然后再去填充列表的略顯繁瑣的操作。

cubes = [i ** 3 for i in range(20)]

看，我們已經(jīng)將三行代碼轉(zhuǎn)換成一行，這無疑是不錯的選擇——你的眼睛不用上下左右的檢查代碼。

把代碼壓縮到一行會使閱讀變得更加困難，但對于推導表達式來說，情況并非如此。您需要的所有元素都被很好地呈現(xiàn)出來，一旦您習慣了語法，它實際上比for循環(huán)版本更具可讀性。

另一點是賦值現(xiàn)在更像是一個原子操作——我們聲明什么是cubes，而不是給出如何構(gòu)建它的指令。這使得代碼讀起來更舒適，因為我們關(guān)心變量cubes是什么，而不是它的構(gòu)造細節(jié)。

最后，表達式通常比在循環(huán)中構(gòu)建集合更快，如果考慮性能，這也是重要因素。

2.用增量賦值替換賦值

增量賦值是一種快速而簡單的Python語法。

只要有這樣的代碼：

count = count + other_value

都可以替換成下面的代碼：

count += other_value

代碼是簡短和清晰的-我們不需要考慮計數(shù)變量兩次。其他可以使用的運算符包括-=， =，/=和 *=。

不過有件事需要稍微小心一點，你要分配給的類型必須定義適當?shù)倪\算符。例如，numpy數(shù)組不支持/=操作。

3.只使用一次的內(nèi)聯(lián)變量

我們在人們的代碼中經(jīng)常看到的一種情況是將結(jié)果賦給臨時變量，然后立即返回它。

def state_attributes(self):
    """Return the state attributes."""
    state_attr = {
        ATTR_CODE_FORMAT: self.code_format,
        ATTR_CHANGED_BY: self.changed_by,
    }
    return state_attr

其實更好的方法是直接返回結(jié)果，而不是再用一個臨時變量存放結(jié)果

def state_attributes(self):
    """Return the state attributes."""
    return {
        ATTR_CODE_FORMAT: self.code_format,
        ATTR_CHANGED_BY: self.changed_by,
    }

這樣可以縮短代碼并刪除不必要的變量，從而減少閱讀代碼的腦力消耗。

臨時變量可能有用的地方是，如果它們被用作參數(shù)或條件，并且名稱可以代表內(nèi)容。在上面的例子中，返回的是state屬性，并且state_attr沒有提供任何額外的信息。因此不必將結(jié)果賦給臨時變量。

4.用if表達式替換if語句

經(jīng)常會遇到的一種情況是，您經(jīng)常希望將一個變量設(shè)置為兩個不同值中的一個。

if condition:
    x = 1
else:
    x = 2

這可以使用Python的條件表達式語法（python的三元運算符版本）在一行上編寫：

x = 1 if condition else 2

這肯定更簡潔，但它是一個更有爭議的重構(gòu)（就像列表表達式）。一些程序員不喜歡這樣的表達式，因為覺得它們比完全寫出if條件更難解析。

我們的觀點是，只要條件表達式很短并且適合合并，那就是一種改進，提升效率。與列表表達式的示例類似，當我們閱讀代碼時，通常不需要知道x是如何分配的，只需看到它被賦值，然后繼續(xù)向前。

5.用生成器代替不需要的表達式

一個技巧是像any、all和sum這樣的函數(shù)允許傳入generator而不是collection。這意味著，與其這樣做：

hat_found = any([is_hat(item) for item in wardrobe])

可以將代碼改成：

hat_found = any(is_hat(item) for item in wardrobe)

這將刪除一對括號，使代碼稍微清晰一些。如果any函數(shù)找到結(jié)果，會立即返回，而不必構(gòu)建整個列表。這可以導致性能提升。

請注意，我們實際上是將生成器傳遞到any（）中，嚴格地說，代碼應(yīng)該如下所示：

hat_found = any(is_hat(item) for item in wardrobe)

但是Python允許您省略這對括號。下面是接受generator的標準庫函數(shù)：

'all', 'any', 'enumerate', 'frozenset', 'list', 'max', 'min', 'set', 'sum', 'tuple'

6.將條件簡化為return語句

最后介紹的重構(gòu)技巧是，函數(shù)需要返回結(jié)果是True或False的情況。一種常見的方法是：

def function():
    if isinstance(a, b) or issubclass(b, a):
        return True
    return False

但是，直接返回結(jié)果會更簡潔，如下所示：

def function():
    return isinstance(a, b) or issubclass(b, a)

只有當表達式的計算結(jié)果為布爾值時，才能這樣操作。例如

def any_hats():
    hats = [item for item in wardrobe if is_hat(item)]
    if hats or self.wearing_hat():
        return True
    return False

這個示例，可以通過bool()將hat和self.wearing_hat()合成bool列表，就可以消除if條件，達到簡化程序的目的。

def any_hats():
    hats = [item for item in wardrobe if is_hat(item)]
    return bool(hats or self.wearing_hat())

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

最新評論