Python小知識之幾種推導式用法示例

更新時間：2023年01月10日 09:44:49 作者：小圓-

Python推導式是一種獨特的數(shù)據(jù)處理方式,可以從一個數(shù)據(jù)序列構建另一個新的數(shù)據(jù)序列的結構體,下面這篇文章主要給大家介紹了關于Python小知識之幾種推導式用法的相關資料,需要的朋友可以參考下

前言

Python 推導式是一種獨特的數(shù)據(jù)處理方式，可以從一個數(shù)據(jù)序列構建另一個新的數(shù)據(jù)序列的結構體。

Python 支持各種數(shù)據(jù)結構的推導式：

列表(list)推導式
字典(dict)推導式
集合(set)推導式
元組(tuple)推導式

1.列表推導式

列表推導式格式為：

[表達式 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：列表生成元素表達式，可以是有返回值的函數(shù)。
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 整除的整數(shù)：

>>> multiples = [i for i in range(30) if i % 3 == 0]
>>> print(multiples)
[0, 3, 6, 9, 12, 15, 18, 21, 24, 27]

2.字典推導式

字典推導基本格式：

{ key_expr: value_expr for value in collection }
或
{ key_expr: value_expr for value in collection if condition }

使用字符串及其長度創(chuàng)建字典：

listdemo = ['Google','Lizexiong', 'Taobao']
# Python學習交流群：770699889
# 將列表中各字符串值為鍵，各字符串的長度為值，組成鍵值對
>>> newdict = {key:len(key) for key in listdemo}
>>> newdict
{'Google': 6, 'Lizexiong': 6, 'Taobao': 6}

提供三個數(shù)字，以三個數(shù)字為鍵，三個數(shù)字的平方為值來創(chuàng)建字典：

>>> dic = {x: x**2 for x in (2, 4, 6)}
>>> dic
{2: 4, 4: 16, 6: 36}
>>> type(dic)
<class 'dict'>

3.集合推導式

集合推導式基本格式：

{ expression for item in Sequence }
或
{ expression for item in Sequence if conditional }

計算數(shù)字 1,2,3 的平方數(shù)：

>>> 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'>

4.元組推導式

元組推導式可以利用 range 區(qū)間、元組、列表、字典和集合等數(shù)據(jù)類型，快速生成一個滿足指定需求的元組。

元組推導式基本格式：

(expression for item in Sequence )
或
(expression for item in Sequence if conditional )

元組推導式和列表推導式的用法也完全相同，只是元組推導式是用 () 圓括號將各部分括起來，而列表推導式用的是中括號 []，另外元組推導式返回的結果是一個生成器對象。

例如，我們可以使用下面的代碼生成一個包含數(shù)字 1~9 的元組：

>>> a = (x for x in range(1,10))
>>> a
<generator object <genexpr> at 0x7faf6ee20a50>  # 返回的是生成器對象

>>> tuple(a)       # 使用 tuple() 函數(shù)，可以直接將生成器對象轉換成元組
(1, 2, 3, 4, 5, 6, 7, 8, 9)