Python 推导式是一种独特的数据处理方式,可以从一个数据序列构建另一个新的数据序列的结构体。
目录
支持各种数据结构的推导式:
- 列表(list)推导式
- 字典(dict)推导式
- 集合(set)推导式
- 元组(tuple)推导式
1.列表推导式
语法格式:[表达式 for 变量 in 列表]
[表达式 for 变量 in 列表 if 条件]
>>> list_1 = [1,2,3,4,5,6]
# 计算列表 list_1中能被2整除的数
>>> list_2 = [i for i in list_1 if i % 2 == 0]
>>> list_2
[2, 4, 6]
# 计算 30 以内可以被 3 整除的整数:
>>> m = [i for i in range(30) if i % 3 == 0]
>>> print(m)
[0, 3, 6, 9, 12, 15, 18, 21, 24, 27]
2.字典推导式
语法格式:
{ key_expr: value_expr for value in collection } 或 { key_expr: value_expr for value in collection if condition }
# 提供三个数字,以三个数字为键,三个数字的平方为值来创建字典:
>>> dic = {x: x**2 for x in (2, 4, 6)}
>>> dic
{2: 4, 4: 16, 6: 36}
>>> type(dic)
<class 'dict'>
3.集合推导式
语法格式:
{ expression for item in Sequence } 或 { expression for item in Sequence if conditional }
# 判断不是 abc 的字母并输出:
>>> a = {x for x in 'abracadabra' if x not in 'abc'}
>>> a
{'d', 'r'}
>>> type(a)
<class 'set'>
# 计算数字 1,2,3 的平方数:
>>> setnew = {i**2 for i in (1,2,3)}
>>> setnew
{1, 4, 9}
4元组推导式
元组推导式可以利用 range 区间、元组、列表、字典和集合等数据类型,快速生成一个满足指定需求的元组。
语法格式:
(expression for item in Sequence ) 或 (expression for item in Sequence if conditional )
元组推导式和列表推导式的用法也完全相同,只是元组推导式是用 () 圆括号将各部分括起来,而列表推导式用的是中括号 [ ],另外元组推导式返回的结果是一个生成器对象。
# 生成一个包含数字 1~9 的元组:
>>> a = (x for x in range(1,10))
>>> a
<generator object <genexpr> at 0x7faf6ee20a50> # 返回的是生成器对象
>>> tuple(a) # 使用 tuple() 函数,可以直接将生成器对象转换成元组
(1, 2, 3, 4, 5, 6, 7, 8, 9)