Python推导式是一种独特的处理方式,可以从一个数据序列构建另一个新的数据序列的结构体。
Python支持的各种数据结构的推导式:
1、列表(list)推导式
2、字典(dict)推导式
3、集合(set)推导式
4、元组(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:条件语句,可以过滤列表中不符合条件的值。
实例解析1:过滤长度小于4的字符串列表,并且将剩下的转换为大写字母:
names = ['Jack', 'Tom', 'justin', 'don', 'jodie', 'Smith', 'vivian']
new_names = [name.upper() for name in names if len(name) >= 4]
print(new_names)
打印结果:
['JACK', 'JUSTIN', 'JODIE', 'SMITH', 'VIVIAN']
实例解析2:计算100以内可以被9整除的整数:
number = [i for i in range(100) if i % 9 == 0]
print(number)
打印结果:
[0, 9, 18, 27, 36, 45, 54, 63, 72, 81, 90, 99]
字典推导式
字典推导式的基本格式:
{ key_expr: value_expr for value in collection }
或者
{ key_expr: value_expr for value in collection if condition }
实例解析1:给出一个字符串,使用该字符串的和长度创建一个新的字典:
names = ['Jack', 'Tom', 'justin', 'don', 'jodie', 'Smith', 'vivian']
newdict = {key: len(key) for key in names}
print(newdict)
打印结果:
{'Jack': 4, 'Tom': 3, 'justin': 6, 'don': 3, 'jodie': 5, 'Smith': 5, 'vivian': 6}
实例解析2:给到三个数字,以这三个数据为key,三个数字的平方作为value来创建字典:
dic = {x: x**2 for x in (1, 2, 3)}
print(dic)
打印结果:
{1: 1, 2: 4, 3: 9}
集合推导式
集合推导式的基本格式:
{ expression for item in Sequence }
或者
{ expression for item in Sequence if conditional }
实例解析1:计算1,2,3,4,5的平方数:
number = {i ** 2 for i in (1, 2, 3, 4, 5)}
print(number)
打印结果:
{1, 4, 9, 16, 25}
实例解析2:判断字符串种不是ab的字母并且输出:
a = {x for x in 'abcdadaaeed' if x not in 'ab'}
print(a)
打印结果:
{'c', 'd', 'e'}
元组推导式
元组推导式可以利用 range 区间、元组、列表、字典和集合等数据类型,快速生成一个满足指定需求的元组。
元组推导式基本格式:
(expression for item in Sequence )
或者
(expression for item in Sequence if conditional )
元组推导式和列表推导式的用法也完全相同,只是元组推导式是用 ( ) 圆括号将各部分括起来,而列表推导式用的是中括号 [ ],另外元组推导式返回的结果是一个生成器对象。
实例解析:生成一个包含数字1-9的元组:
a = (x for x in range(1, 10))
print(a)
print(tuple(a))
打印结果:
<generator object <genexpr> at 0x000001EA6BB6CBA0>
(1, 2, 3, 4, 5, 6, 7, 8, 9)