推导式将for循环与创建新元素的代码合并成一行,只要你掌握了推导式的规则,你就可以用一行代码生成满足条件的列表、字典、集合。
- 列表推导式
- 字典推导式
- 集合推导式
一、列表推导式
列表推导式总共有两种形式:
① list= [exp1 for x in data if condition]
此处if主要起条件判断作用,data数据中只有满足if条件的才会被留下,并计算exp1的值,最后统一生成为一个数据列表
② list = [exp1 if condition else exp2 for x in data]
此处if…else主要起赋值作用,当data中的数据满足if条件时将其做exp1处理,否则按照exp2处理,最后统一生成为一个数据列表
注意点:
1:list 是最后要得到的列表的名字
2:用 []
3:exp是关于变量的表达式或返回值
4:在①中if进行判断,在②中if进行赋值选择
例:
>>> even = [ i for i in range(1,100) if i %2 == 0]
>>> print(even)
[2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28, 30, 32, 34, 36, 38, 40, 42, 44, 46, 48, 50, 52, 54, 56, 58, 60, 62, 64, 66, 68, 70, 72, 74, 76, 78, 80, 82, 84, 86, 88, 90, 92, 94, 96, 98]
二、字典推导式
字典推导和列表推导的使用方法是类似的,只不中括号该改成大括号。
{ key_expr: value_expr for value in collection if condition }