上QQ阅读APP看书,第一时间看更新
4.4 迭代进阶
4.4.1 Iterable
上一节介绍了for循环,这种遍历叫作迭代,在Python中并不是所有对象都可以进行迭代的,例如对一个整数进行迭代就会抛出异常:
输出结果:
错误信息表明整数对象是不可以进行迭代操作的。如何判断一个对象是否可以进行迭代呢?可以使用collections模块的Iterable类型来判断:
输出:
>>> False >>> True
4.4.2 enumerate
enumerate函数可以将一个序列转换为索引-元素对,方便在操作序列时使用元素的索引。下面改进for循环,使用enumerate函数在循环中同时迭代索引和元素本身:
for i, value in enumerate(["Aaron", "Tom", "Kate"]): print(i, value)
4.4.3 列表推导式
列表推导式(list comprehension)是利用其他列表创建新列表的一种方式。它的工作方式类似于for循环,例如:
>>> [x*x for x in range(10)] >>> [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
上面列表推导式中的“for x in range(10)”相当于一个for循环,range()函数用来创建一个新的列表,在每次循环的时候将循环变量x赋值给表达式x*x,x*x相当于for循环的代码块,最后根据x*x生成新列表。
我们知道for循环是可以嵌套if条件判断语句的,在列表推导式中一样可以,例如,生成一个奇数数组:
>>> [x for x in range(10) if x%2 != 0] >>> [1, 3, 5, 7, 9]
当然列表推导式中的for循环也是可以嵌套的,例如:
>>> [(x, y) for x in range(3) for y in ["Aaron", "Tom", "Jack"]] >>> [(0, 'Aaron'), (0, 'Tom'), (0, 'Jack'), (1, 'Aaron'), (1, 'Tom'), (1, 'Jack'), (2, 'Aaron'), (2, 'Tom'), (2, 'Jack')]
注意
虽然列表推导式非常灵活,但是为了保证代码的易读性,建议在列表推导式中不要嵌套过多层次,推荐最多两层。