1.列表推導(dǎo)式

列表推導(dǎo)式提供了一個(gè)更簡單的創(chuàng)建列表的方法。常見的用法是把某種操作應(yīng)用于序列或可迭代對象的每個(gè)元素上，然后使用其結(jié)果來創(chuàng)建列表，或者通過滿足某些特定條件元素來創(chuàng)建子序列。

例如，假設(shè)我們想創(chuàng)建一個(gè)平方列表，像這樣

>>> squares = [] >>> for x in range(10): ... squares.append(x**2) ... >>> squares [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

注意這里創(chuàng)建（或被重寫）的名為 x 的變量在for循環(huán)后仍然存在。我們可以計(jì)算平方列表的值而不會產(chǎn)生任何副作用

squares = list(map(lambda x: x**2, range(10)))

或者，等價(jià)于

squares = [x**2 for x in range(10)]

上面這種寫法更加簡潔易讀。

Python列表推導(dǎo)式的結(jié)構(gòu)是由一對方括號所包含的以下內(nèi)容：一個(gè)表達(dá)式，后面跟一個(gè) for 子句，然后是零個(gè)或多個(gè) for或 if 子句。 其結(jié)果將是一個(gè)新列表，由對表達(dá)式依據(jù)后面的 for 和 if 子句的內(nèi)容進(jìn)行求值計(jì)算而得出。 舉例來說，以下列表推導(dǎo)式會將兩個(gè)列表中不相等的元素組合起來:

>>> [(x, y) for x in [1,2,3] for y in [3,1,4] if x != y] [(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]

而它等價(jià)于

>>> combs = [] >>> for x in [1,2,3]: ... for y in [3,1,4]: ... if x != y: ... combs.append((x, y)) ... >>> combs [(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]

注意在上面兩個(gè)代碼片段中， for 和 if 的順序是相同的。

如果表達(dá)式是一個(gè)Python元組（例如上面的 (x, y)），那么就必須加上括號

>>> vec = [-4, -2, 0, 2, 4] >>> # create a new list with the values doubled >>> [x*2 for x in vec] [-8, -4, 0, 4, 8] >>> # filter the list to exclude negative numbers >>> [x for x in vec if x >= 0] [0, 2, 4] >>> # apply a function to all the elements >>> [abs(x) for x in vec] [4, 2, 0, 2, 4] >>> # call a method on each element >>> freshfruit = [' banana', ' loganberry ', 'passion fruit '] >>> [weapon.strip() for weapon in freshfruit] ['banana', 'loganberry', 'passion fruit'] >>> # create a list of 2-tuples like (number, square) >>> [(x, x**2) for x in range(6)] [(0, 0), (1, 1), (2, 4), (3, 9), (4, 16), (5, 25)] >>> # the tuple must be parenthesized, otherwise an error is raised >>> [x, x**2 for x in range(6)]File "<stdin>", line 1, in <module>[x, x**2 for x in range(6)]^ SyntaxError: invalid syntax >>> # flatten a list using a listcomp with two 'for' >>> vec = [[1,2,3], [4,5,6], [7,8,9]] >>> [num for elem in vec for num in elem] [1, 2, 3, 4, 5, 6, 7, 8, 9]

列表推導(dǎo)式可以使用復(fù)雜的表達(dá)式和嵌套函數(shù)

>>> from math import pi >>> [str(round(pi, i)) for i in range(1, 6)] ['3.1', '3.14', '3.142', '3.1416', '3.14159']

2.嵌套的列表推導(dǎo)式

列表推導(dǎo)式中的初始表達(dá)式可以是任何表達(dá)式，包括另一個(gè)列表推導(dǎo)式。

考慮下面這個(gè) 3x4的矩陣，它由3個(gè)長度為4的列表組成

>>> matrix = [ ... [1, 2, 3, 4], ... [5, 6, 7, 8], ... [9, 10, 11, 12], ... ]

下面的列表推導(dǎo)式將交換其行和列

>>> [[row[i] for row in matrix] for i in range(4)] [[1, 5, 9], [2, 6, 10], [3, 7, 11], [4, 8, 12]]

如上節(jié)所示，嵌套的列表推導(dǎo)式是基于跟隨其后的 for 進(jìn)行求值的，所以這個(gè)例子等價(jià)于:

>>> transposed = [] >>> for i in range(4): ... transposed.append([row[i] for row in matrix]) ... >>> transposed [[1, 5, 9], [2, 6, 10], [3, 7, 11], [4, 8, 12]]

反過來說，也等價(jià)于

''' 學(xué)習(xí)中遇到問題沒人解答？小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流QQ群：857662006 尋找有志同道合的小伙伴，互幫互助,群里還有不錯(cuò)的視頻學(xué)習(xí)教程和PDF電子書！ ''' >>> transposed = [] >>> for i in range(4): ... # the following 3 lines implement the nested listcomp ... transposed_row = [] ... for row in matrix: ... transposed_row.append(row[i]) ... transposed.append(transposed_row) ... >>> transposed [[1, 5, 9], [2, 6, 10], [3, 7, 11], [4, 8, 12]]

實(shí)際應(yīng)用中，你應(yīng)該會更喜歡使用內(nèi)置函數(shù)去組成復(fù)雜的流程語句。 zip() 函數(shù)將會很好地處理這種情況

>>> list(zip(*matrix)) [(1, 5, 9), (2, 6, 10), (3, 7, 11), (4, 8, 12)]

總結(jié)

以上是生活随笔為你收集整理的Python教程：列表推导式和嵌套的列表推导式讲解的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。