在示例 10-16 中,为了实现 Vector 类,我用了几个生成器表达式,__eq____hash____abs__angleanglesformat__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 类示例说明了如何把生成器当作典型的迭代器使用,即从集合中获取元素。不过,生成器也可用于生成不受数据源限制的值。下一节会举例说明。