總結(jié)python列表推導(dǎo)和生成器表達(dá)式

小編這次要給大家分享的是總結(jié)python列表推導(dǎo)和生成器表達(dá)式，文章內(nèi)容豐富，感興趣的小伙伴可以來了解一下，希望大家閱讀完這篇文章之后能夠有所收獲。

首先來看一下代碼：

chars = "abcd"
tmp = []
for char in chars:
 tmp.append(ord(char))
print(tmp)

這是一般的寫法，將字符串中的每一個字符轉(zhuǎn)換稱ASCII碼，然后存進(jìn)一個tmp數(shù)組。

利用列表推導(dǎo)的方式：

tmp = [ord(char) for char in chars]

輸出：[97,98,99,100]

列表推導(dǎo)中也可以使用if和else。

當(dāng)只有一個if時：

chars = "abcd"
tmp = [ord(char) for char in chars if ord(char)>=98]
print(tmp)

輸出：['b','c','d']

額外提一下，當(dāng)然我們也可以利用filter和map達(dá)到相同的效果：

tmp = list(filter(lambda x:x>=98,map(ord,chars)))

這句話的意思是，使用map對chars中的每一個字符利用ord函數(shù)進(jìn)行處理。再使用filter對這每一個字符的ASCII碼進(jìn)行過濾大于等于98的，最后將這些符合條件的值存進(jìn)一個列表中。lambda x:x>98是一個匿名函數(shù)。

有一個if else時：

chars = "abcd"
tmp = [True if ord(char)>=98 else False for char in chars]
print(tmp)

輸出：[False, True, True, True]

一般而言，列表推導(dǎo)應(yīng)盡量保持簡短。

再看下一個例子：利用列表推導(dǎo)計算笛卡兒積（以衣服的顏色和尺寸為例）

colors = ["black","white"]
sizes = ["S","M","L"]
tshirts = [(color,size) for color in colors for size in sizes]
print(tshirts)

生成器表達(dá)式可以用于生成其它數(shù)據(jù)類型的序列，雖然列表推導(dǎo)也可以初始化元祖、數(shù)組和其它類型，但是生成器表達(dá)式是更好的選擇。這是因為生成器表達(dá)式背后遵從迭代器協(xié)議，可以逐個產(chǎn)生元素，而不是先建立一個完整的列表。

生成器表達(dá)式語法和列表差不多，只不過把方括號變成圓括號。

用生成器表達(dá)式生成元組和數(shù)組例子：

chars = "abcd"
tmp = tuple(ord(char) for char in chars)
print(tmp)

輸出：(97, 98, 99, 100)

import array
chars = "abcd"
a = array.array("I",(ord(char) for char in chars))
print(a)

輸出：array('I', [97, 98, 99, 100])

array的構(gòu)造方法須提供兩個參數(shù)，第一個是數(shù)組中元素的存儲方式。如果生成器表達(dá)式是唯一的參數(shù)，就不用括號括起來，這里不是唯一的，因此要用括號括起來。

colors = ["black","white"]
sizes = ["S","M","L"]
for tshirt in ("%s %s" % (c,s) for c in colors for s in sizes):
  print(tshirt)

輸出：

black S
black M
black L
white S
white M
white L

看完這篇關(guān)于總結(jié)python列表推導(dǎo)和生成器表達(dá)式的文章，如果覺得文章內(nèi)容寫得不錯的話，可以把它分享出去給更多人看到。