Python 可迭代的对象、迭代器 和生成器(何时使用生成器表达式)
何时使用生成器表达式
在示例 10-16 中,为了实现 Vector 类,我用了几个生成器表达
式,eq、hash、abs、angle、angles、format、add
和 __mul__
方法中各有一个生成器表达式。在这些方法中使用列表推
导也行,不过立即返回的列表要使用更多的内存。
通过示例 14-9 可知,生成器表达式是创建生成器的简洁句法,这样无
需先定义函数再调用。不过,生成器函数灵活得多,可以使用多个语句
实现复杂的逻辑,也可以作为协程使用(参见第 16 章)。
遇到简单的情况时,可以使用生成器表达式,因为这样扫一眼就知道代
码的作用,如 Vector 类的示例所示。
根据我的经验,选择使用哪种句法很容易判断:如果生成器表达式要分
成多行写,我倾向于定义生成器函数,以便提高可读性。此外,生成器
函数有名称,因此可以重用。
句法提示
如果函数或构造方法只有一个参数,传入生成器表达式时不用写一
对调用函数的括号,再写一对括号围住生成器表达式,只写一对括
号就行了,如示例 10-16 中 __mul__
方法对 Vector 构造方法的调
用,转摘如下。然而,如果生成器表达式后面还有其他参数,那么
必须使用括号围住,否则会抛出 SyntaxError 异常:
def __mul__(self, scalar):if isinstance(scalar, numbers.Real):return Vector(n * scalar for n in self)else:return NotImplemented
目前所见的 Sentence 类示例说明了如何把生成器当作典型的迭代器使
用,即从集合中获取元素。不过,生成器也可用于生成不受数据源限制的值。下一节会举例说明。