推導(dǎo)式
推導(dǎo)式是從一個或者多個迭代器快速簡潔地創(chuàng)建數(shù)據(jù)結(jié)構(gòu)的一種方法。它可以
將循環(huán)和條件判斷結(jié)合,從而避免語法冗長的代碼。
可以結(jié)合range() 函數(shù)使用一個迭代器:
>>> number_list = []
>>> for number in range(1, 6):
... number_list.append(number)
...
>>> number_list
[1, 2, 3, 4, 5]
num_list = list(range(0,10))
print(num_list)
上面這些方法都是可行的Python 代碼,會得到相同的結(jié)果。然而,更像Python 風(fēng)格的創(chuàng)
建列表方式是使用列表推導(dǎo)。最簡單的形式如下所示:
[ expression for item in iterable ]
>>> number_list = [number for number in range(1,6)]
>>> number_list
[1, 2, 3, 4, 5]
########################
>>> number_list = [number-1 for number in range(1,6)]
>>> number_list
[0, 1, 2, 3, 4]
列表推導(dǎo)也可以像下面的例子加上條件表達(dá)式:
[expression for item in iterable if condition]
>>> a_list = [a for a in range(0,11) if a%2 ==0]
>>> a_list
[0, 2, 4, 6, 8, 10]
>>> a_list = [a-10 for a in range(1,11) if a%2 ==0]
>>> a_list
[-8, -6, -4, -2, 0]
####################################
rows = range(1,4)
cols = range(1,3)
>>> for row in rows:
... for col in cols:
... print(row,col)
...
1 1
1 2
2 1
2 2
3 1
3 2
#########################################
>>> cells =[(row,col) for row in rows for col in cols]
>>> print(cells)
[(1, 1), (1, 2), (2, 1), (2, 2), (3, 1), (3, 2)]
>>> for cell in cells:
... print(cell)
...
(1, 1)
(1, 2)
(2, 1)
(2, 2)
(3, 1)
(3, 2)
字典推導(dǎo)式
除了列表,字典也有自己的推導(dǎo)式。最簡單的例子就像:
{ key_expression : value_expression for expression in iterable }
類似于列表推導(dǎo),字典推導(dǎo)也有if 條件判斷以及多個for 循環(huán)迭代語句:
>>> word = 'letters'
>>> letter_counts = {letter: word.count(letter) for letter in word}
>>> letter_counts
{'l': 1, 'e': 2, 't': 2, 'r': 1, 's': 1}
集合推導(dǎo)式
集合也不例外,同樣有推導(dǎo)式。最簡單的版本和之前的列表、字典推導(dǎo)類似:
{expression for expression in iterable }
最長的版本(if tests, multiple for clauses)對于集合而言也是可行的:
>>> a_set = {number for number in range(1,6) if number % 3 == 1}
>>> a_set
{1, 4}
生成器推導(dǎo)式
元組是沒有推導(dǎo)式的。你可能認(rèn)為將列表推導(dǎo)式中的方括號變成圓括號就可以定義
元組推導(dǎo)式,就像下面的表達(dá)式一樣:
>>> number_thing = (number for number in range(1, 6))
其實,圓括號之間的是生成器推導(dǎo)式,它返回的是一個生成器對象:
>>> type(number_thing)
<class 'generotor’>