Python 推導式是一種獨特的資料處理方式,可以從一個資料序列建構另一個新的資料序列的結構體。
Python 支援各種資料結構的推導式:
- 清單(list)推導式
- 字典(dict)推導式
- 集合(set)推導式
- 元組(tuple)推導式
清單推導式
清單推導式格式為:
[表達式 for 變量 in 清單]
[out_exp_res for out_exp in input_list]
或者
[表達式 for 變量 in 清單 if 條件]
[out_exp_res for out_exp in input_list if condition]
- out_exp_res:清單生成元素表達式,可以是有傳回值的函數。
- for out_exp in input_list:疊代 input_list 将 out_exp 傳入到 out_exp_res 表達式中。
- if condition:條件語句,可以過濾清單中不符合條件的值。
過濾掉長度小于或等于3的字元串清單,并将剩下的轉換成大寫字母:
執行個體
>>> names = ['Bob','Tom','alice','Jerry','Wendy','Smith']
>>> new_names = [name.upper()for name in names if len(name)>3]
>>> print(new_names)
['ALICE', 'JERRY', 'WENDY', 'SMITH']
計算 30 以内可以被 3 整除的整數:
>>> multiples = [i for i in range(30) if i % 3 == 0]
>>> print(multiples)
[0, 3, 6, 9, 12, 15, 18, 21, 24, 27]
字典推導式
字典推導基本格式:
{ key_expr: value_expr for value in collection }
或
{ key_expr: value_expr for value in collection if condition }
使用字元串及其長度建立字典:
listdemo = ['Google','Runoob', 'Taobao']
# 将清單中各字元串值為鍵,各字元串的長度為值,組成鍵值對
>>> newdict = {key:len(key) for key in listdemo}
>>> newdict
{'Google': 6, 'Runoob': 6, 'Taobao': 6}
提供三個數字,以三個數字為鍵,三個數字的平方為值來建立字典:
>>> dic = {x: x**2 for x in (2, 4, 6)}
>>> dic
{2: 4, 4: 16, 6: 36}
>>> type(dic)
<class 'dict'>
集合推導式
集合推導式基本格式:
{ expression for item in Sequence }
或
{ expression for item in Sequence if conditional }
計算數字 1,2,3 的平方數:
>>> setnew = {i**2 for i in (1,2,3)}
>>> setnew
{1, 4, 9}
判斷不是 abc 的字母并輸出:
>>> a = {x for x in 'abracadabra' if x not in 'abc'}
>>> a
{'d', 'r'}
>>> type(a)
<class 'set'>
元組推導式
元組推導式可以利用 range 區間、元組、清單、字典和集合等資料類型,快速生成一個滿足指定需求的元組。
元組推導式基本格式:
(expression for item in Sequence )
或
(expression for item in Sequence if conditional )
元組推導式和清單推導式的用法也完全相同,隻是元組推導式是用 () 圓括号将各部分括起來,而清單推導式用的是中括号 [],另外元組推導式傳回的結果是一個生成器對象。
例如,我們可以使用下面的代碼生成一個包含數字 1~9 的元組:
>>> a = (x for x in range(1,10))
<generator object <genexpr> at 0x7faf6ee20a50> # 傳回的是生成器對象
>>> tuple(a) # 使用 tuple() 函數,可以直接将生成器對象轉換成元組
(1, 2, 3, 4, 5, 6, 7, 8, 9)