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的字符串列表,并将剩下的转换成大写字母:
lst = [1, 2, 3, 4, 5]
lst_new = [i * i for i in lst if i > 2]
print(lst_new)
# [9, 16, 25]
字典推导基本格式:
{ key_expr: value_expr for value in collection }
或
{ key_expr: value_expr for value in collection if condition }
提供三个数字,以三个数字为键,三个数字的平方为值来创建字典:
new_dic = {x: x**2 for x in (2, 4, 6)}
print(new_dic)
#结果:{2: 4, 4: 16, 6: 36}
集合推导式基本格式:
{ expression for item in Sequence }
或
{ expression for item in Sequence if conditional }
判断不是 abc 的字母并输出新的集合:
new_set = {x for x in 'abracadabra' if x not in 'abc'}
print(new_set)
#结果:{'d', 'r'} 集合中元素不重复,会自动去重
元组推导式可以利用 range 区间、元组、列表、字典和集合等数据类型,快速生成一个满足指定需求的元组。
元组推导式基本格式:
(expression for item in Sequence )
或
(expression for item in Sequence if conditional )
元组推导式和列表推导式的用法也完全相同,只是元组推导式是用 () 圆括号将各部分括起来,而列表推导式用的是中括号 [],另外元组推导式返回的结果是一个生成器对象。
例如,判断不是 abc 的字母并输出新的元组:
new_tup = (x for x in 'abracadabra' if x not in 'abc')
print(new_tup) # # 返回的是生成器对象
print(tuple(new_tup)) # 使用 tuple() 函数,可以直接将生成器对象转换成元组
"""
结果:
<generator object <genexpr> at 0x018B16F0>
('r', 'd', 'r')
"""