Python 推导式是一种独特的数据处理方式,可以从一个数据序列构建另一个新的数据序列的结构体。
Python 推导式是一种强大且简洁的语法,适用于生成列表、字典、集合和生成器。
在使用推导式时,需要注意可读性,尽量保持表达式简洁,以免影响代码的可读性和可维护性。
列表(list)推导式
列表推导式允许你根据已有的列表创建一个新列表
列表推导式格式为:
[表达式 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]
'''
字典(dict)推导式
字典推导式允许通过已有的可迭代对象快速构建字典,其中包含键值对。
字典推导基本格式:
{ key_expr: value_expr for value in collection } 或 { key_expr: value_expr for value in collection if condition }
# 使用字符串及其长度创建字典:
listdemo = ['Google', 'CSDN', 'Taobao']
# 将列表中各字符串值为键,各字符串的长度为值,组成键值对
newdict = {key: len(key) for key in listdemo}
print(newdict)
'''
{'Google': 6, 'CSDN': 4, 'Taobao': 6}
'''
# 提供三个数字,以三个数字为键,三个数字的平方为值来创建字典:
dic = {x: x ** 2 for x in (2, 4, 6)}
print(dic)
print(type(dic))
'''
{2: 4, 4: 16, 6: 36}
<class 'dict'>
'''
# 交换字典的键值对
my_dict = {'a': 1, 'b': 2, 'c': 3}
flipped_dict = {value: key for key, value in my_dict.items()}
print(flipped_dict)
'''
{1: 'a', 2: 'b', 3: 'c'}
'''
集合(set)推导式
集合推导式与列表推导式类似,但是结果是一个集合而不是列表。
集合推导式基本格式:
{ expression for item in Sequence } 或 { expression for item in Sequence if conditional }
# 计算数字 1,2,3 的平方数:
setnew = {i ** 2 for i in (1, 2, 3)}
print(setnew)
'''
{1, 4, 9}
'''
# 判断不是 abc 的字母并输出:
a = {x for x in 'abracadabra' if x not in 'abc'}
print(a)
print(type(a))
'''
{'r', 'd'}
<class 'set'>
'''
元组(tuple)推导式
在 Python 中,元组推导式(Tuple Comprehension)实际上是不存在的。Python 中并没有直接支持元组推导式的语法,与列表推导式、集合推导式和字典推导式不同。
虽然没有元组推导式,但可以使用生成器表达式来生成元组。生成器表达式类似于推导式,但它返回一个生成器对象而不是列表、集合或字典。如果需要将生成器表达式的结果转换为元组,可以使用 tuple()
函数来完成。元组推导式和列表推导式的用法也完全相同,只是元组推导式是用 () 圆括号将各部分括起来,而列表推导式用的是中括号 []
元组推导式基本格式:
(expression for item in Sequence ) 或 (expression for item in Sequence if conditional )
# 我们可以使用下面的代码生成一个包含数字 1~9 的元组:
a = (x for x in range(1,10))
print(a)
'''
<generator object <genexpr> at 0x7faf6ee20a50>
#返回的是生成器对象
'''
print(tuple(a))
'''
(1, 2, 3, 4, 5, 6, 7, 8, 9)
'''
# 创建一个包含整数的元组,这些整数是某些条件下的平方数:
# 使用生成器表达式生成包含平方数的元组
squares_tuple = tuple(x**2 for x in range(5))
print(squares_tuple)
'''
(0, 1, 4, 9, 16)
'''
注意事项:
可读性和复杂性:推导式的一大优点是简洁性,但是过度复杂的推导式可能会降低代码的可读性。确保推导式不要变得太长或者嵌套层次太深,以免影响代码的可维护性。
避免副作用:在推导式中,避免使用会改变状态或有副作用的操作,比如修改外部变量或者进行I/O操作。推导式应该保持简单、纯粹,只处理数据转换和筛选。
条件和过滤器的使用:当使用条件表达式来过滤数据时,确保条件清晰明了。复杂的条件可以分解成多个步骤或使用函数来提高可读性。
适应性:虽然推导式非常强大,但并不是所有情况都适合使用推导式。在某些情况下,传统的循环语句可能更清晰和易于理解。
总之,Python推导式是一种强大的语法工具,能够简化代码并提高效率。但是,在使用时要注意上述注意事项,以确保代码的可读性、性能和可维护性。