文章目錄

• 一、Python 推導式
• 二、列表推導式
• 三、字典推導式
• 四、集合推導式
• 五、元組推導式
• 參考鏈接

一、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))
>>> a
<generator object <genexpr> at 0x7faf6ee20a50> # 返回的是生成器對象
>>> tuple(a) # 使用 tuple() 函數，可以直接將生成器對象轉換成元組
(1, 2, 3, 4, 5, 6, 7, 8, 9)

參考鏈接

1. Python 推導式