
推導(dǎo)式是Python中按照固定的條件表達(dá)式,自動(dòng)生成數(shù)據(jù)的一種語法規(guī)則。
通過推導(dǎo)式,可以直接優(yōu)化傳統(tǒng)的通過循環(huán)結(jié)構(gòu)和選擇結(jié)構(gòu)實(shí)現(xiàn)的列表操作,簡化傳統(tǒng)語法生成數(shù)據(jù)的規(guī)則或者過濾數(shù)據(jù)的規(guī)則。
目前,Python存在著三種推導(dǎo)式,分別為:列表推導(dǎo)式、字典推導(dǎo)式和集合推導(dǎo)式。其中,最常用的就是列表推導(dǎo)式。
列表推導(dǎo)式
推導(dǎo)式的概念并不是很難理解,所以說直接看案例代碼,可以更加熟練。
首先,我們先來看一個(gè)簡單的案例,用傳統(tǒng)的列表循環(huán)方式實(shí)現(xiàn)數(shù)字1-10的平方。
num_list = list()
for i in range(1,11):
num_list.append(i**2)
print(num_list)
# [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
接下來使用推導(dǎo)式實(shí)現(xiàn)數(shù)字1-10的平方。
num_list = [x**2 for x in range(1,11)]
print(num_list)
# [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
通過代碼的對比,我們可以發(fā)現(xiàn),推導(dǎo)式語法確實(shí)十分簡潔優(yōu)雅,只需要一行的代碼就可以實(shí)現(xiàn)Python的循環(huán)結(jié)構(gòu)或選擇結(jié)構(gòu),基本上任何可迭代的數(shù)據(jù)都可以在推導(dǎo)式內(nèi)進(jìn)行更新。
string = 'i love python'
# 將字符串按照空格分割成一個(gè)列表
lst = [ i for i in string.split(' ')]
print(lst)
# ['i', 'love', 'python']
當(dāng)我們需要一些稍顯復(fù)雜的數(shù)據(jù)時(shí),此時(shí)需要對推導(dǎo)式內(nèi)的表達(dá)式進(jìn)行嵌套,以便得到我們想要的數(shù)據(jù)。
lst = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
# 獲取列表lst中的偶數(shù)
num_list = [j for i in lst for j in i if j % 2 == 0]
print(num_list)
# [2, 4, 6, 8]
由上述代碼可見,一旦推導(dǎo)式開始嵌套,可讀性便會(huì)稍微有所下降,所以,我們需要對上述代碼進(jìn)行優(yōu)化。
lst = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
num_list = [
j
for i in lst
for j in i
if j % 2 == 0
]
通過上述代碼的改寫,是不是直觀了許多。
字典推導(dǎo)式
字典推導(dǎo)式原理上跟列表推導(dǎo)式其實(shí)并無不同,只需要將方括號[ ]更換為花括號{ },換湯不換藥而已。
dct = {1: 'a', 2: 'b', 3: 'c'}
# key與value的互換
dict_list = {value: key for key, value in dct.items()}
print(dict_list)
# {'a': 1, 'b': 2, 'c': 3}
集合推導(dǎo)式
集合推導(dǎo)式和字典推導(dǎo)式一樣,用的都是花括號{ },在使用時(shí),需要依靠變量和循環(huán)體對字典推導(dǎo)式進(jìn)行區(qū)分。
集合沒有順序,不可重復(fù),配合推導(dǎo)式有時(shí)會(huì)有一些不錯(cuò)的效果。
lst = [1, 2, 2, 3, 4, 5, 6, 6, 7, 8, 8, 9]
# 獲取偶數(shù)項(xiàng)
sets = {i for i in lst if i % 2 == 0}
print(sets)
# {8, 2, 4, 6}
不過呢,我們需要記住,推導(dǎo)式只是一種簡化語法,不可盲目追求這種寫法,畢竟傳統(tǒng)語法的可讀性往往更友好,并且,對推導(dǎo)式進(jìn)行排錯(cuò)會(huì)更加困難。