对生成器和迭代器的理解（2）

什么是迭代器

迭代是访问集合元素的一种方式。迭代器是一个可以记住遍历的位置的对象。迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问完结束。迭代器只能往前不会后退。

什么是集合元素

list、tuple、str等类型的数据

什么是迭代

可以对list、tuple、str等类型的数据使用for…in…的循环语法从其中依次拿到数据进行使用，我们把这样的过程称为遍历，也叫迭代。

但是，是否所有的数据类型都可以放到for…in…的语句中，然后让for…in…每次从中取出一条数据供我们使用，即供我们迭代吗？

测试列表

    for temp in [11,22,33]:
        print(temp)

    # 测试元组
    for temp in ("A","B","C"):
        print(temp)

    # 测试字典
    person_info={"name":"xiaowang","age":20}
    for item in  person_info.items():
        print(item)

    # 测试字符串
    for temp in "itcast":
        print(temp)

    # 测试整数
    for temp in 100:
        print(temp)

运行结果：

11
22
33
A
B
C
('name', 'xiaowang')
('age', 20)
i
t
c
a
s
t
Traceback (most recent call last):
  File "/home/python/Desktop/demo2/cc2/shengcheng/demo01.py", line 19, in <module>
    for temp in 100:
TypeError: 'int' object is not iterable

Process finished with exit code 1

我们自定义一个容器MyList用来存放数据，可以通过add方法向其中添加数据

from collections import Iterable


class Mylist(object):
    def __init__(self):
        self.items = []

    def add(self,item):
        self.items.append(item)

my_list = Mylist()
my_list.add('xiaowang')
my_list.add('xiaoli')
my_list.add('xiaozhang')
my_list.add('xiaofu')

print(my_list)

print(isinstance(my_list,Iterable))

for temp in my_list:
    print(temp)

运行结果：

<__main__.Mylist object at 0x7f26e70d59b0>
False
Traceback (most recent call last):
  File "/home/python/Desktop/demo2/cc2/shengcheng/demo01.py", line 21, in <module>
    for temp in my_list:
TypeError: 'Mylist' object is not iterable

Process finished with exit code 1

我们自定义了一个容器类型MyList，在将一个存放了多个数据的MyList对象放到for…in…的语句中，发现for…in…并不能从中依次取出一条数据返回给我们，也就说我们随便封装了一个可以存放多条数据的类型却并不能被迭代使用。

我们把可以通过for…in…这类语句迭代读取一条数据供我们使用的对象称之为可迭代对象（Iterable）**。

如何判断一个对象是否可迭代

可以使用 isinstance() 判断一个对象是否是 Iterable 对象：

from collections import Iterable

# 判断列表
print(isinstance([], Iterable))

# 判断元组
print(isinstance((), Iterable))

# 判断字典
print(isinstance({}, Iterable))

# 判断字符串
print(isinstance("", Iterable))

# 判断整数
print(isinstance(100, Iterable))

# 判断boolean
print(isinstance(False, Iterable))

可迭代对象的本质

我们分析对可迭代对象进行迭代使用的过程，发现每迭代一次（即在for…in…中每循环一次）都会返回对象中的下一条数据，一直向后读取数据直到迭代了所有数据后结束。那么，在这个过程中就应该有一个“人”去记录每次访问到了第几条数据，以便每次迭代都可以返回下一条数据。我们把这个能帮助我们进行数据迭代的“人”称为迭代器(Iterator)。

可迭代对象的本质就是可以向我们提供一个这样的中间“人”即迭代器帮助我们对其进行迭代遍历使用。

可迭代对象通过_iter_方法向我们提供一个迭代器，我们在迭代一个可迭代对象的时候，实际上就是先获取该对象提供的一个迭代器，然后通过这个迭代器来依次获取对象中的每一个数据.

那么也就是说，一个具备了iter方法的对象，就是一个可迭代对象。

from collections import Iterable


class Mylist(object):
    def __init__(self):
        self.items = []

    def add(self,item):
        self.items.append(item)

    def __iter__(self):
        pass

my_list = Mylist()
my_list.add('xiaowang')
my_list.add('xiaoli')
my_list.add('xiaozhang')
my_list.add('xiaofu')

print(my_list)

print(isinstance(my_list,Iterable))

for temp in my_list:
    print(temp)

运行结果：
<__main__.Mylist object at 0x7f2fcda2c9b0>
True
Traceback (most recent call last):
  File "/home/python/Desktop/demo2/cc2/shengcheng/demo01.py", line 24, in <module>
    for temp in my_list:
TypeError: iter() returned non-iterator of type 'NoneType'

Process finished with exit code 1

这回测试发现添加了iter方法的mylist对象已经是一个可迭代对象了……

iter()函数与next()函数

list、tuple等都是可迭代对象，我们可以通过iter()函数获取这些可迭代对象的迭代器。然后我们可以对获取到的迭代器不断使用next()函数来获取下一条数据。iter()函数实际上就是调用了可迭代对象的iter方法。

li = [11, 22, 33, 44, 55]

# 通过iter() 取得可迭代对象的迭代器
iterator = iter(li)

# 通过next()函数取得iterator迭代器指向的下一个值
print(next(iterator))
print(next(iterator))
print(next(iterator))
print(next(iterator))
print(next(iterator))
print(next(iterator))

运行结果：

11
22
33
44
55
Traceback (most recent call last):
  File "/home/python/Desktop/demo2/cc2/shengcheng/demo01.py", line 11, in <module>
    print(next(iterator))
StopIteration

Process finished with exit code 1

迭代器Iterator

通过上面的分析，我们已经知道，迭代器是用来帮助我们记录每次迭代访问到的位置，当我们对迭代器使用next()函数的时候，迭代器会向我们返回它所记录位置的下一个位置的数据。

实际上，在使用next()函数的时候，调用的就是迭代器对象的_next_方法（Python3中是对象的next方法，Python2中是对象的next()方法）。

所以，我们要想构造一个迭代器，就要实现它的_next_方法。但这还不够，python要求迭代器本身也是可迭代的，所以我们还要为迭代器实现_iter_方法，而_iter_方法要返回一个迭代器，迭代器自身正是一个迭代器，所以迭代器的_iter_方法返回自身即可。

一个实现了iter方法和next方法的对象，就是迭代器。

import collections


class MyList(object):
    def __init__(self):
        self.items = []

    def add(self, item):
        self.items.append(item)

    def __iter__(self):
        """ 返回一个迭代器Iterator"""
        iterator = MyIterator(self)
        return iterator


class MyIterator(object):
    """自定义供上面可迭代对象使用的一个迭代器"""

    def __init__(self, list):
        self.mylist = list
        # current 用来记录当前访问到的位置
        self.current = 0

    def __next__(self):
        if self.current < len(self.mylist.items):
            item = self.mylist.items[self.current]
            self.current += 1
            return item
        else:
            raise StopIteration

    def __iter__(self):
        return self


myList = MyList()
myList.add("小王")
myList.add("小张")
myList.add("小李")

# 测试自定义的对象是否为可迭代对象
print(isinstance(myList, collections.Iterable))

# 遍历
for name in myList:
    print(name)

for…in…循环的本质

for item in Iterable 循环的本质就是先通过iter()函数获取可迭代对象Iterable的迭代器，然后对获取到的迭代器不断调用next()方法来获取下一个值并将其赋值给item，当遇到StopIteration的异常后循环结束。

迭代器的应用场景

我们发现迭代器最核心的功能就是可以通过next()函数的调用来返回下一个数据值。如果每次返回的数据值不是在一个已有的数据集合中读取的，而是通过程序按照一定的规律计算生成的，那么也就意味着可以不用再依赖一个已有的数据集合，也就是说不用再将所有要迭代的数据都一次性缓存下来供后续依次读取，这样可以节省大量的存储（内存）空间。
举个例子，比如，数学中有个著名的斐波拉契数列（Fibonacci），数列中第一个数为0，第二个数为1，其后的每一个数都可由前两个数相加得到：

0, 1, 1, 2, 3, 5, 8, 13, 21, 34, …

现在我们想要通过for…in…循环来遍历迭代斐波那契数列中的前n个数。那么这个斐波那契数列我们就可以用迭代器来实现，每次迭代都通过数学计算来生成下一个数。

class Fib(object):
    def __init__(self,n):
        self.n = n
        self.current = 0
        self.num1 = 0
        self.num2 = 1
    def __next__(self):
        if self.current < self.n:
            num = self.num1
            self.num1,self.num2 = self.num2,self.num1+self.num2
            self.current += 1
            return num
        else:
            raise StopIteration
    def __iter__(self):
        # 定义了__iter__ 说明是可迭代的
        # __iter__返回一个迭代器 （迭代器内部实现__next__方法）
        # 并且这个迭代器本身也是可迭代的 具有__iter__方法
        return self

if __name__ == '__main__':
    fib = Fib(10)
    for num in fib:
        print('  ',num,end = '')

运行结果：

   0   1   1   2   3   5   8   13   21   34
Process finished with exit code 0