列表推導式與字典推導式
在 Python 中推導式是一種非常 Pythonic
的知識眶俩,本篇博客將為你詳細解答列表推導式與字典推導式相關的技術知識。
列表推導式
列表推導式可以利用列表鸠珠,元組杯矩,字典沽损,集合等數(shù)據(jù)類型亮元,快速的生成一個特定需要的列表猛计。
語法格式如下:
[表達式 for 迭代變量 in 可迭代對象 [if 條件表達式]]
if 條件表達式
非必選,學完列表推導式之后爆捞,你可以發(fā)現(xiàn)它就是 for 循環(huán)的一個變種語句奉瘤,例如咱們現(xiàn)有一個需求是將一個列表中的所有元素都變成原值的 2 倍。
for 循環(huán)寫法
my_list = [1,2,3]
new_list = []
for i in my_list:
new_list.append(i*2)
print(new_list)
列表推導式寫法
nn_list = [i*2 for i in my_list]
print(nn_list)
是不是對比看就是將 for 循環(huán)語句做了變形之后煮甥,增加了一個 []
盗温,不過需要注意的是,列表推導式最終會將得到的各個結果組成一個新的列表成肘。
再看一下列表推導式語法構成 nn_list = [i*2 for i in my_list]
卖局,for
關鍵字后面就是一個普通的循環(huán),前面的表達式 i*2
其中的 i
就是 for
循環(huán)中的變量双霍,也就是說表達式可以用后面 for
循環(huán)迭代產(chǎn)生的變量砚偶,理解這個內(nèi)容列表推導式就已經(jīng)掌握 9 成內(nèi)容了批销,剩下的是熟練度的問題。
在將 if
語句包含進代碼中蟹演,運行之后风钻,你也能掌握基本技巧顷蟀,if
語句是一個判斷酒请,其中 i
也是前面循環(huán)產(chǎn)生的迭代變量。
nn_list = [i*2 for i in my_list if i>1]
print(nn_list)
這些都是一般技能鸣个,列表推導式能支持兩層 for
循環(huán)羞反,例如下述代碼:
nn_list = [(x,y) for x in range(3) for y in range(3) ]
print(nn_list)
當然如果你想加密(誰都看不懂你的代碼)你的代碼,你可以無限套娃下去囤萤,列表推導式并沒有限制循環(huán)層數(shù)昼窗,多層循環(huán)就是一層一層的嵌套,你可以展開一個三層的列表推導式涛舍,就都明白了
nn_list = [(x,y,z,m) for x in range(3) for y in range(3) for z in range(3) for m in range(3)]
print(nn_list)
當然在多層列表推導式里面澄惊,依舊支持 if 語句
,并且 if
后面可以用前面所有迭代產(chǎn)生的變量富雅,不過不建議超過 2 成掸驱,超過之后會大幅度降低你代碼的可閱讀性。
當然如果你希望你代碼更加難讀没佑,下面的寫法都是正確的毕贼。
nn_list = [(x, y, z, m) for x in range(3) if x > 1 for y in range(3) if y > 1 for z in range(3) for m in range(3)]
print(nn_list)
nn_list = [(x, y, z, m) for x in range(3) for y in range(3) for z in range(3) for m in range(3) if x > 1 and y > 1]
print(nn_list)
nn_list = [(x, y, z, m) for x in range(3) for y in range(3) for z in range(3) for m in range(3) if x > 1 if y > 1]
print(nn_list)
現(xiàn)在你已經(jīng)對列表推導式有比較直觀的概念了,列表推導式對應的英文是 list comprehension
蛤奢,有的地方寫作列表解析式鬼癣,基于它最后的結果,它是一種創(chuàng)建列表的語法啤贩,并且是很簡潔的語法待秃。
有了兩種不同的寫法,那咱們必須要對比一下效率痹屹,經(jīng)測試小數(shù)據(jù)范圍影響不大锥余,當循環(huán)次數(shù)到千萬級時候,出現(xiàn)了一些差異痢掠。
import time
def demo1():
new_list = []
for i in range(10000000):
new_list.append(i*2)
def demo2():
new_list = [i*2 for i in range(10000000)]
s_time = time.perf_counter()
demo2()
e_time = time.perf_counter()
print("代碼運行時間:", e_time-s_time)
運行結果:
# for 循環(huán)
代碼運行時間: 1.3431036140000001
# 列表推導式
代碼運行時間: 0.9749278849999999
在 Python3 中列表推導式具備局部作用域驱犹,表達式內(nèi)部的變量和賦值只在局部起作用,表達式的上下文里的同名變量還可以被正常引用足画,局部變量并不會影響到它們雄驹。所以其不會有變量泄漏的問題。例如下述代碼:
x = 6
my_var = [x*2 for x in range(3)]
print(my_var)
print(x)
列表推導式還支持嵌套
參考代碼如下淹辞,只有想不到医舆,沒有做不到的。
my_var = [y*4 for y in [x*2 for x in range(3)]]
print(my_var)
字典推導式
有了列表推導式的概念,字典推導式學起來就非常簡單了蔬将,語法格式如下:
{鍵:值 for 迭代變量 in 可迭代對象 [if 條件表達式]}
直接看案例即可
my_dict = {key: value for key in range(3) for value in range(2)}
print(my_dict)
得到的結果如下:
{0: 1, 1: 1, 2: 1}
此時需要注意的是字典中不能出現(xiàn)同名的 key爷速,第二次出現(xiàn)就把第一個值覆蓋掉了,所以得到的 value 都是 1霞怀。
最常見的哪里還是下述的代碼惫东,遍歷一個具有鍵值關系的可迭代對象。
my_tuple_list = [('name', '橡皮擦'), ('age', 18),('class', 'no1'), ('like', 'python')]
my_dict = {key: value for key, value in my_tuple_list}
print(my_dict)
元組推導式與集合推導式
其實你應該能猜到毙石,在 Python 中是具備這兩種推導式的廉沮,而且語法相信你已經(jīng)掌握了。不過語法雖然差不多徐矩,但是元組推導式運行結果卻不同滞时,具體如下。
my_tuple = (i for i in range(10))
print(my_tuple)
運行之后產(chǎn)生的結果:
<generator object <genexpr> at 0x0000000001DE45E8>
使用元組推導式生成的結果并不是一個元組滤灯,而是一個生成器對象坪稽,需要特別注意下,這種寫法在有的地方會把它叫做生成器語法鳞骤,不叫做元組推導式窒百。
集合推導式也有一個需要注意的地方,先看代碼:
my_set = {value for value in 'HelloWorld'}
print(my_set)
因為集合是無序且不重復的弟孟,所以會自動去掉重復的元素贝咙,并且每次運行顯示的順序不一樣,使用的時候很容易暈掉拂募。
這篇博客的總結
這篇博客庭猩,我們學習了列表和字典推導式,掌握并熟練的應用二者之后陈症,你 Python 技能又向前進了一步蔼水。