学会python列表生成式,让你的代码“骚”起来
列表生成式
• 顾名思义,列表生成式就是一个用来生成列表的特定语法形式的表达式。
例如我们需要生成一个列表,由0到9组成,那我们就可以这样写:
list1 = [0,1,2,3,4,5,6,7,8,9]
感觉,手动写这10个数字还好,但是,有时候,测试算法或者函数什么的,这样手动去写,是不是太累了??有没有更加方便的方法.?有.
首先,我们先介绍一下range.
• range有什么作用
- range() 函数可创建一个整数列表,一般用在 for循环中。
函数语法 range(start, stop[, step])
参数说明:
• start: 计数从 start 开始。默认是从 0 开始。例如range(5)等价于range(0, 5);
• stop: 计数到 stop 结束,但不包括 stop。例如:range(0, 5) 是[0, 1, 2, 3, 4]没有5
• step:步长,默认为1。例如:range(0, 5) 等价于 range(0, 5, 1) 实例
range(10) # 从 0 开始到 10
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
range(1, 11) # 从 1 开始到 11
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
range(0, 30, 5) # 步长为 5
[0, 5, 10, 15, 20, 25]
range(0, 10, 3) # 步长为 3
[0, 3, 6, 9]
range(0, -10, -1) # 负数
[0, -1, -2, -3, -4, -5, -6, -7, -8, -9]
range(0)
[]
range(1, 0)
[]
通过使用range,可以快速生成一个有顺序的列表
然后,事情并没有那么简单. 假如,我需要给列表里面每个元素进行特殊处理,例如是加减法的话,又如何处理呢?
- 用for重新操作一次.
- 划重点.用列表生成式.
基础语法格式 [exp for iter_var in iterable]
工作过程:
• 迭代iterable中的每个元素;
• 每次迭代都先把结果赋值给iter_var,然后通过exp得到一个新的计算值;
• 最后把所有通过exp得到的计算值以一个新列表的形式返回。
相当于这样的过程:
L = []
for iter_var in iterable:
L.append(exp)
循环嵌套语法格式 [exp for iter_var_A in iterable_A for iter_var_B in iterable_B]
工作过程:
• 每迭代iterable_A中的一个元素,就把ierable_B中的所有元素都迭代一遍。
相当于这样的过程:
L = []
for iter_var_A in iterable_A:
for iter_var_B in iterable_B:
L.append(exp)