第十四天02生成器千锋python人工智能学院.pdfVIP

生成器 1. 生成器 利用迭代器，我们可以在每次迭代获取数据（通过next()方法）时按照特定的规律进行生成。但是我们在实现 一个迭代器时，关于当前迭代到的状态需要我们自己记录，进而才能根据当前状态生成下一个数据。为了达到 记录当前状态，并配合next()函数进行迭代使用，我们可以采用更简便的语法，即生成器(generator) 。生成 器是⼀类特殊的迭代器。 2. 创建生成器方法1 要创建一个生成器，有很多种方法。第 法很简单，只要把一个列表生成式的 [ ] 改成 ( ) In [15]: L = [ x*2 for x in range(5)] In [16]: L Out[16]: [0, 2, 4, 6, 8] In [17]: G = ( x*2 for x in range(5)) 创建 L 和 G 的区别仅在于最外层的 [ ] 和 ( ) ， L 是一个列表，而 G 是一个生成器。我们可以直接打印出列 表L的每一个元素，而对于生成器G，我们可以按照迭代器的使用方法来使用，即可以通过next()函数、for循 环、list()等方法使用。 In [19]: next(G) Out[19]: 0 In [20]: next(G) Out[20]: 2 In [21]: next(G) Out[21]: 4 In [22]: next(G) Out[22]: 6 In [23]: next(G) Out[23]: 8 In [24]: next(G) StopI tion Traceback (most recent call last) ipython-input-24-380e167d6934 in module() 1 next(G) StopI tion: In [25]: In [26]: G = ( x*2 for x in range(5)) In [27]: for x in G: : print(x) : 0 2 4 6 8 In [28]: 3. 创建生成器方法2 generator非常强大。如果推算的算法比较复杂，用类似列表生成式的 for 循环无法实现的时候，还可以用函 数来实现。 我们仍然用上一节提到的斐波那契数列来举例，回想我们在上一节用迭代器的实现方式： class FibI tor(object): 那契数列迭代器 def init (self, n): :param n: int, 指明生成数列的前n个数 self.n = n # current用来保存当前生成到数列中的第几个数了 self.current = 0 # num1用来保存前前一个数，初始值为数列中的第一个数0 self.num1 = 0 # num2用来保存前一个数，初始值为数列中的第二个数1 def next (self): 被next()函数调用来获取下一个数 if self.current self.n: num = self.num1 self.num1, self.num2 = self.num2