Django 2.0 入门与实践
上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')]

注意

虽然列表推导式非常灵活,但是为了保证代码的易读性,建议在列表推导式中不要嵌套过多层次,推荐最多两层。