Python基础(十六):迭代器与生成器

:point_down: :point_down: :point_down: 看下面,Python系列知识在此

原文链接:Python 基础(十六):迭代器与生成器_Python小二-CSDN博客
原文作者:Python小二

1 迭代器

迭代

我们知道 Python 中有一些对象可以通过 for 来循环遍历,比如:列表、元组、字符等,以字符串为例,如下所示:

for i in 'Hello':

    print(i)

执行结果:

H

e

l

l

o

这个遍历过程就是迭代。

可迭代对象

可迭代对象需具有 __iter__() 方法,它们均可使用 for 循环遍历,我们可以使用 isinstance() 方法来判断一个对象是否为可迭代对象,看下示例:

from collections import Iterable

print(isinstance('abc', Iterable))

print(isinstance({1, 2, 3}, Iterable))

print(isinstance(1024, Iterable))

执行结果:

True

True

False

迭代器

迭代器需要具有 __iter__()__next__() 两个方法,这两个方法共同组成了迭代器协议,通俗来讲迭代器就是一个可以记住遍历位置的对象,迭代器一定是可迭代的,反之不成立。

  • __iter__():返回迭代器对象本身
  • __next__():返回下一项数据

迭代器对象本质是一个数据流,它通过不断调用 __next__() 方法或被内置的 next() 方法调用返回下一项数据,当没有下一项数据时抛出 StopIteration 异常迭代结束。上面我们说的 for 循环语句的实现便是利用了迭代器。

我们试着自己来实现一个迭代器,如下所示:

class MyIterator:

    def __init__(self):

        self.s = '程序之间'

        self.i = 0

    def __iter__(self):

        return self

    def __next__(self):

        if self.i < 4:

            n = self.s[self.i]

            self.i += 1

            return n

        else:

            raise StopIteration

mi = iter(MyIterator())

for i in mi:

    print(i)

输出结果:

程

序

之

间

2 生成器

生成器是用来创建迭代器的工具,其写法与标准函数类似,不同之处在于返回时使用 yield 语句,关于 yield ,我们在使用 Scrapy 爬取去哪儿网景区信息中已经作了一些介绍,我们再来熟悉一下:

yield 是一个关键字,作用和 return 差不多,差别在于 yield 返回的是一个生成器(在 Python 中,一边循环一边计算的机制,称为生成器),它的作用是:有利于减小服务器资源,在列表中所有数据存入内存,而生成器相当于一种方法而不是具体的信息,用多少取多少,占用内存小。

生成器的创建方式有很多种,比如:使用 yield 语句、生成器表达式(可以简单的理解为是将列表的 [] 换成了 (),特点是更加简洁,但不够灵活)。看下示例:

示例 1

def reverse(data):

    for i in range(len(data)-1, -1, -1):

        yield data[i]

for char in reverse('Hello'):

    print(char)

执行结果:

o

l

l

e

H

示例 2

# 列表

lis = [x*x for x in range(5)]

print(lis)

# 生成器

gen = (x*x for x in range(5))

for g in gen:

    print(g)

执行结果:

[0, 1, 4, 9, 16]

0

1

4

9

16

参考:
https://docs.python.org/zh-cn/3/tutorial/classes.html#iterators