實用炫酷！這樣寫Python程式碼四兩撥千斤！

Python技術

判斷一個程式設計師水平的高低，不能光看他的髮量，也不能光看他的程式碼量，還要看他程式碼蘊含的思想，程式碼的質量。程式碼蘊含的思想主要體現在各種設計模式的運用上，而程式碼的質量就既要實現需求，又要保證程式碼的簡潔優雅。保證程式碼質量是需要長期積累，養成良好的程式設計習慣，不斷思考最佳化的。

今天就給大家介紹一種實用的保證程式碼簡潔的武功—— Python 推導四式。

什麼是推導式

推導式 comprehensions（又稱解析式），是 python 的一種獨有特性。推導式是可以從一個數據序列構建另一個新的資料序列。

看定義很抽象，下面我們來看看具體的例子，透過例子來學習這四種推導式。

列表推導式

我們先來看一個需求：

快速建立一個包含元素1-9的平方的列表

面對這個需求，我們通常的實現方式是這樣的：

lis = ［］

for i in range（1， 10）：

lis。append（i*i）

print（lis） # ［1， 4， 9， 16， 25， 36， 49， 64， 81］

用列表推導式怎麼實現呢？你只需要一行程式碼：

lis = ［x * x for x in range（1， 10）］

print（list） # ［1， 4， 9， 16， 25， 36， 49， 64， 81］

這就是推導式。我們可以看到列推導式的語法是這樣的：

變數名 = ［表示式 for 變數 in 列表］

再來一個複雜點的：

list = ［x * y for x in range（1， 10） for y in range（1， 10）］

print（lis）

# ［1， 2， 3， 4， 5， 6， 7， 8， 9， 2， 4， 6， 8， 10， 12， 14， 16， 18， 3， 6， 9， 12， 15， 18， 21， 24， 27， 4， 8， 12， 16， 20， 24， 28， 32， 36， 5， 10， 15， 20， 25， 30， 35， 40， 45， 6， 12， 18， 24， 30， 36， 42， 48， 54， 7， 14， 21， 28， 35， 42， 49， 56， 63， 8， 16， 24， 32， 40， 48， 56， 64， 72， 9， 18， 27， 36， 45， 54， 63， 72， 81］

這個是多個變數的型別，因此我們可以擴充套件上面的語法：

變數名 = ［表示式 for 變數 in 列表 for 變數 in xxx］

當然，還有一種帶條件的列推導式，它的語法是這樣的：

變數名 = ［表示式 for 變數 in 列表 if 條件］

我們來看一個需求：快速建立一個包含1-10之間所有偶數的列表。

lis = ［i for i in range（1， 11） if i % 2 == 0］

print（lis） # ［2， 4， 6， 8， 10］

字典推導式

字典推導式的形式和列表推導式類似，語法也是類似的，只不過字典推導式返回的結果是字典。

變數名 = {key： value表示式}

我們來看一個例子：

dic = {x： x/2 for x in range（1，11） if x % 2 == 0}

print（dic） # {2： 1。0， 4： 2。0， 6： 3。0， 8： 4。0， 10： 5。0}

如果這裡我寫成下面這種方式，返回的結果是什麼呢？

dic = {‘half’： x/2 for x in range（1，11） if x % 2 == 0}

print（dic）

大家可以自己嘗試執行一下，看跟你想的結果是不是一樣的。

集合推導式

看了前面兩個推導式，想必你已經知道集合推導式怎麼寫了。語法如下：

變數名 = {表示式 for 變數 in 列表 for 變數 in xxx}

或者

變數名 = {表示式 for 變數 in 列表 if 條件}

來一個實際的例子：建立一個儲存10個偶數的集合。

set1 = {x for x in range（10） if x % 2 == 0}

print（set1） # {0， 2， 4， 6， 8}

元組推導式

我們先來看一個例子：

tup=（x for x in range（1，10））

print（tup） # at 0x1101fade0>

看到這個結果列印是不是有點懵？

上面的程式碼返回的變數其實是一個生成器，並不是一個元組。其實沒有真正的元組推導式，我們只能用一個類似的方法來生成元組，暫且叫做“偽元組推導式”吧。

上面的例子我們來改進一下：

tup=tuple（x for x in range（1，10））

print（tup） # （1， 2， 3， 4， 5， 6， 7， 8， 9）

在前面加上 tuple 來顯式進行型別轉換就行了。

總結

這裡介紹了四種 python 的推導式，主要是用來簡化 迴圈的程式碼，生成不同的資料結構用的。當然，從這些基礎的表示式語法出發，還可以運用到複雜的推導式，大家在寫程式碼時留心使用就會發現它的強大之處。這種一行程式碼既簡潔明瞭，又可以在新手面前炫技，用起來不要太爽！