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：条件语句，可以过滤列表中不符合条件的值。

实例

将一个列表里的元素选出长度超过三的，并将其全部大写

in_list = ['apple','Tom','sun','keyboard','python']
new_list = [i.upper() for i in in_list if len(i)>3]
print(new_list)

['APPLE', 'KEYBOARD', 'PYTHON']

进程已结束,退出代码0

 

输出100以内的偶数

new_list = [i for i in range(100) if i%2==0]
print(new_list)

[0, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28, 30, 32, 34, 36, 38, 40, 42, 44, 46, 48, 50, 52, 54, 56, 58, 60, 62, 64, 66, 68, 70, 72, 74, 76, 78, 80, 82, 84, 86, 88, 90, 92, 94, 96, 98]

进程已结束,退出代码0

二、字典的推导式

字典推导基本格式：

{ key_expr: value_expr for value in collection }

或

{ key_expr: value_expr for value in collection if condition }

实例，以列表中的字符串为键，字符串的长度为值，组成字典

new_dict = {key:len(key) for key in ['apple','keyboard','python']}
#使用字符串当字典的健，字符串的长度当字典的值，组成键值对
print(new_dict)

{'apple': 5, 'keyboard': 8, 'python': 6}

进程已结束,退出代码0

三、集合推导式

集合推导式基本格式：

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

实例：

将字符串中满足条件的放入集合中，最后输出集合

_exp = {i for i in "abshushuihuis" if i in 'acd'}
print(_exp)
print(*_exp)
print(type(*_exp))
print(type(_exp))

{'a'}
a
<class 'str'>
<class 'set'>

进程已结束,退出代码0

四、元组的推导式（生成器的表达式）

元组推导式可以利用 range 区间、元组、列表、字典和集合等数据类型，快速生成一个满足指定需求的元组。

元组推导式基本格式：

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

元组推导式和列表推导式的用法也完全相同，只是元组推导式是用 () 圆括号将各部分括起来，而列表推导式用的是中括号 []，另外元组推导式返回的结果是一个生成器对象。

实例：

生成一个100以内的3的倍数的元组

_tuple = (key for key in range(100) if key %3==0)
print(_tuple)#返回的是生成器对象
print(tuple(_tuple))#使用tuple()强制转换为元组

<generator object <genexpr> at 0x000001660C91A448>
(0, 3, 6, 9, 12, 15, 18, 21, 24, 27, 30, 33, 36, 39, 42, 45, 48, 51, 54, 57, 60, 63, 66, 69, 72, 75, 78, 81, 84, 87, 90, 93, 96, 99)

进程已结束,退出代码0

通过学习了这几种推导式，可以使我们在代码的编写过程中更加便利，能够优化代码