天天看點

Python 推導式

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)