Python 炫技操作：推導(dǎo)式的五種寫(xiě)法

推導(dǎo)式是可以從一個(gè)數(shù)據(jù)序列構(gòu)建另一個(gè)新的數(shù)據(jù)序列的結(jié)構(gòu)體。

總共有四種推導(dǎo)式：

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

2. 字典(dict)推導(dǎo)式

3. 集合(set)推導(dǎo)式

4. 生成器推導(dǎo)式

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

列表推導(dǎo)式的基本格式

new_list?=?[expression?for_loop_expression?if?condition]

舉個(gè)例子。

我想找出一個(gè)數(shù)值列表中為偶數(shù)的元素，并組成新列表，通常不用列表推導(dǎo)式，可以這么寫(xiě)

old_list?=?[0,1,2,3,4,5]

new_list?=?[]
for?item?in?old_list:
????if?item?%?2?==?0:
????????new_list.append(item)

print(new_list)?#?output:?[0,?2,?4]

一個(gè)簡(jiǎn)單的功能，寫(xiě)的代碼倒是不少。

如果使用了列表推導(dǎo)式，那就簡(jiǎn)潔多了，而且代碼還變得更加易讀了。

>>>?old_list?=?[0,1,2,3,4,5]
>>>
>>>?new_list?=?[item?for?item?in?old_list?if?item?%?2?==?0]
>>>?print(new_list)?#?output:?[0,?2,?4]
[0,?2,?4]

2. 字典推導(dǎo)式

字典推導(dǎo)式的基本格式，和 列表推導(dǎo)式相似，只是把 [] 改成了 {}，并且組成元素有兩個(gè)：key 和 value，要用 key_expr: value_expr 表示。

new_dict?={?key_expr:?value_expr?for_loop_expression?if?condition?}

舉個(gè)例子。

我想從一個(gè)包含所有學(xué)生成績(jī)信息的字典中，找出數(shù)學(xué)考滿分的同學(xué)。

old_student_score_info?=?{
????"Jack":?{
????????"chinese":?87,
????????"math":?92,
????????"english":?78
????},
????"Tom":?{
????????"chinese":?92,
????????"math":?100,
????????"english":?89
????}
}

new_student_score_info?=?{name:?scores?for?name,?scores?in?old_student_score_info.items()?if?scores["math"]?==?100}
print(new_student_score_info)
#?output:?{'Tom':?{'chinese':?92,?'math':?100,?'english':?89}}

3. 集合推導(dǎo)式

集合推導(dǎo)式跟列表推導(dǎo)式也是類似的。唯一的區(qū)別在于它使用大括號(hào){}，組成元素也只要一個(gè)。

基本格式

new_set?=?{?expr?for_loop_expression?if?condition?}

舉個(gè)例子

我想把一個(gè)數(shù)值列表里的數(shù)進(jìn)行去重處理

>>>?old_list?=?[0,0,0,1,2,3]
>>>
>>>?new_set?=?{item?for?item?in?old_list}
>>>?print(new_set)
{0,?1,?2,?3}

4. 生成器推導(dǎo)式

生成器推導(dǎo)式跟列表推導(dǎo)式，非常的像，只是把 [] 換成了 ()

• 列表推導(dǎo)式：生成的是新的列表

• 生成器推導(dǎo)式：生成的是一個(gè)生成器

直接上案例了，找出一個(gè)數(shù)值列表中所有的偶數(shù)

>>>?old_list?=?[0,1,2,3,4,5]
>>>?new_list?=?(item?for?item?in?old_list?if?item?%?2?==?0)
>>>?new_list
?at?0x10292df10>
>>>?next(new_list)
0
>>>?next(new_list)
2

5. 嵌套推導(dǎo)式

for 循環(huán)可以有兩層，甚至更多層，同樣的，上面所有的推導(dǎo)式，其實(shí)都可以寫(xiě)成嵌套的多層推導(dǎo)式。

但建議最多嵌套兩層，最多的話，代碼就會(huì)變得非常難以理解。

舉個(gè)例子。

我想打印一個(gè)乘法表，使用兩個(gè)for可以這樣寫(xiě)

for?i?in?range(1,?10):
????for?j?in?range(1,?i+1):
????????print('{}x{}={}\t'.format(j,?i,?i*j),?end='')
????print("")

輸出如下

1x1=1????
1x2=2????2x2=4???
1x3=3????2x3=6???3x3=9???
1x4=4????2x4=8???3x4=12??4x4=16??
1x5=5????2x5=10??3x5=15??4x5=20??5x5=25??
1x6=6????2x6=12??3x6=18??4x6=24??5x6=30??6x6=36??
1x7=7????2x7=14??3x7=21??4x7=28??5x7=35??6x7=42??7x7=49??
1x8=8????2x8=16??3x8=24??4x8=32??5x8=40??6x8=48??7x8=56??8x8=64??
1x9=9????2x9=18??3x9=27??4x9=36??5x9=45??6x9=54??7x9=63??8x9=72??9x9=81

如果使用嵌套的列表推導(dǎo)式，可以這么寫(xiě)

>>>?print('\n'.join(['?'.join(['%2d?*%2d?=?%2d'?%?(col,?row,?col?*?row)?for?col?in?range(1,?row?+?1)])?for?row?in?range(1,?10)]))
?1?*?1?=??1
?1?*?2?=??2??2?*?2?=??4
?1?*?3?=??3??2?*?3?=??6??3?*?3?=??9
?1?*?4?=??4??2?*?4?=??8??3?*?4?=?12??4?*?4?=?16
?1?*?5?=??5??2?*?5?=?10??3?*?5?=?15??4?*?5?=?20??5?*?5?=?25
?1?*?6?=??6??2?*?6?=?12??3?*?6?=?18??4?*?6?=?24??5?*?6?=?30??6?*?6?=?36
?1?*?7?=??7??2?*?7?=?14??3?*?7?=?21??4?*?7?=?28??5?*?7?=?35??6?*?7?=?42??7?*?7?=?49
?1?*?8?=??8??2?*?8?=?16??3?*?8?=?24??4?*?8?=?32??5?*?8?=?40??6?*?8?=?48??7?*?8?=?56??8?*?8?=?64
?1?*?9?=??9??2?*?9?=?18??3?*?9?=?27??4?*?9?=?36??5?*?9?=?45??6?*?9?=?54??7?*?9?=?63??8?*?9?=?72??9?*?9?=?81