Python 推导式是一种独特的数据处理方式,可以从一个数据序列构建另一个新的数据序列的结构体。
Python 支持各种数据结构的推导式:
- 列表(list)推导式
- 字典(dict)推导式
- 集合(set)推导式
- 元组(tuple)推导式
注意:没有字符串推导式哦
列表推导式
格式
[表达式 for 变量 in 列表]
[表达式 for 变量 in 列表 if 条件]
简单理解一下,表达式后面的部分就是从原列表中取出元素,通过表达式进行操作,返回操作后的列表
实例
【实例1】过滤掉长度小于或等于3的字符串列表,并将剩下的转换成大写字母
#coding=utf-8
#过滤掉长度小于或等于3的字符串列表,并将剩下的转换成大写字母
names = ['Bob', 'Tom', 'alice', 'Jerry', 'Wendy', 'Smith']
print(names)
new_names = [name.upper()for name in names if len(name) > 3]
print(new_names)
运行结果
['Bob', 'Tom', 'alice', 'Jerry', 'Wendy', 'Smith']
['ALICE', 'JERRY', 'WENDY', 'SMITH']
【实例2】 计算 30 以内可以被 3 整除的整数
#coding=utf-8
#计算 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 }
格式上基本和列表推导式基本没什么区别
实例
【实例1】使用字符串及其长度创建字典
# coding=utf-8
#使用字符串及其长度创建字典
listdemo = ['Google', 'Baidu', 'Tencent']
# 将列表中各字符串值为键,各字符串的长度为值,组成键值对
newdict = {key: len(key) for key in listdemo}
print(newdict)
运行结果:{'Google': 6, 'Baidu': 5, 'Tencent': 7}
【实例2】提供三个数字,以三个数字为键,三个数字的平方为值来创建字典
# coding=utf-8
#提供三个数字,以三个数字为键,三个数字的平方为值来创建字典
dic = {x: x**2 for x in (2, 4, 6)}
print(dic)
运行结果:{2: 4, 4: 16, 6: 36}
集合推导式
格式
{expression for item in Sequence}
{expression for item in Sequence if conditional}
实例
【实例1】计算数字的平方数
# coding=utf-8
#计算数字-1,1,2,3 的平方数
setnew = {i**2 for i in (-1, 1, 2, 3)}
print(setnew)
运行结果:{1, 4, 9}
【实例2】判断不是 abc 的字母并输出
# coding=utf-8
判断不是 abc 的字母并输出
a = {x for x in 'abracadabra' if x not in 'abc'}
print(a)
运行结果:{'d', 'r'}
由于集合元素独一无二 ,使用集合类型能够过滤掉重复元素。我们可以利用集合推导式的这一特性进行数据处理。
元组推导式
元组推导式可以利用 range 区间、元组、列表、字典和集合等数据类型,快速生成一个满足指定需求的元组。
(expression for item in Sequence )
(expression for item in Sequence if conditional )
元组推导式和列表推导式的用法也完全相同,只是元组推导式是用 () 圆括号将各部分括起来,而列表推导式用的是中括号 [],另外元组推导式返回的结果是一个生成器对象。
例如,我们可以使用下面的代码生成一个包含数字 1~9 的元组:
#coding=utf-8
a = (x for x in range(1, 10))
print(a) # 返回的是生成器对象
print(tuple(a)) # 使用 tuple() 函数,可以直接将生成器对象转换成元组
运行结果:
<generator object <genexpr> at 0x000001C2CA5B9E70>
(1, 2, 3, 4, 5, 6, 7, 8, 9)
Python推导式的用法都差不多,重要的是理解它的概念。