Python_迭代器

迭代器

1.迭代

1.1什么是迭代

迭代是访问集合元素的一种方式

例如：

nums = [11, 22, 33]

# 可以通过for循环将nums列表中的每个数据依次获取
for num in nums:
   print(num)
   
   
name = 'teacher'

for temp in name:
   print(temp)

运行效果：

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

2.可迭代对象

只要是可以通过for…in…的形式进行遍历的，那么这个数据类型就是可以迭代的。

例如：
下面的是可以迭代的数据类型

• 列表
• 元组
• 字典
• 字符串

下面的则是不可以进行迭代的数据类型

• 整型
• 浮点型

只要通过isinstance来判断出是Iterable类的实例。即 isinstance的结果是True那么就表示这个数据是可以迭代的数据类型

In [1]: from collections.abc import Iterable

In [2]: isinstance([], Iterable)
Out[2]: True

In [3]: isinstance({}, Iterable)
Out[3]: True

In [4]: isinstance(100, Iterable)
Out[4]: False

3.迭代器

迭代器是一个可以记住遍历的位置的对象。迭代器从第一个元素开始访问，知道所有的元素被访问完结束。迭代器执行只能往前不会后退。

from collections.abc import Iterator


nums = [11, 22, 33, 44]
nums_iter = iter(nums)

print("nums", isinstance(nums, Iterator))
print("nums_iter", isinstance(nums_iter, Iterator))


num1 = next(nums_iter)
print(num1)

num2 = next(nums_iter)
print(num2)

num3 = next(nums_iter)
print(num3)

num4 = next(nums_iter)
print(num4)

num5 = next(nums_iter)  # 这里会产生异常
print(num5)

4. 自定义“迭代器”

4.1 __iter__方法

一个具备了__iter__方法的对象，就是一个可迭代对象。

from collections.abc import Iterable


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

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


mylist = MyList()
mylist.add(11)
mylist.add(22)
mylist.add(33)

# 判断mylist是否是可迭代对象
print(isinstance(mylist, Iterable))
``
效果：
![在这里插入图片描述](https://img-blog.csdnimg.cn/20200806083735815.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0MjEzNjUz,size_16,color_FFFFFF,t_70)
添加__iter__方法后：
```python from collections.abc import Iterable


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

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

    def __iter__(self):
        pass

mylist = MyList()
mylist.add(11)
mylist.add(22)
mylist.add(33)

# 判断mylist是否是可迭代对象
print(isinstance(mylist, Iterable))

效果：

总结：
一个类只要有**iter**方法，那么这个类创建出来的对象就是可以迭代对象
其实，当我们调用iter()函数提取一个可迭代对象的迭代器是，实际上会自动调用这个对象的__iter__方法，并且这个方法返回迭代器。

4.2__next__方法

我们已经知道迭代器是用来帮助我们记录每次迭代访问到的位置。当迭代器使用next()函数的时候，迭代器会向我们返回它所记录位置的下一个位置的数据。实际上，在使用next（）函数的时候调用的是迭代器对象__next__方法（python3中），所以构造一个迭代器没救需要实现__next__方法。因为python要求迭代器本身也是可以迭代的，所以我们需要实现__iter__方法，而且__iter_方法要返回一个迭代器。
所以，一个实现了__iter__方法和__next__方法的对象，就是迭代器。

 
如何判断一个对象是否是迭代器
可以使用isinstance（）判断一个对象是否是Iterator对象：

In [1]: from collections.abc import Iterator

In [2]: isinstance([], Iterator)
Out[2]: False

In [3]: isinstance(iter([]), Iterator)
Out[3]: True

In [4]: isinstance(iter("abc"), Iterator)
Out[4]: True

 

4.2自定义迭代器

from collections.abc import Iterable
from collections.abc import Iterator


class MyList(object):
    """
    自定义的一个可迭代对象
    """

    def __init__(self):
        self.items = []

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

    def __iter__(self):
        return MyIterator()


class MyIterator:
    """
    自定义的共上面可迭代的对象使用
    """
    def __init__(self):
        pass

    def __iter__(self):
        pass

    def __next__(self):
        pass


mylist = MyList()
mylist_iter = iter(mylist)

print("mylist是否是可以迭代对象", isinstance(mylist, Iterable))
print("mylist是否是迭代器", isinstance(mylist, Iterator))

print("mylist_iter是否是可以迭代对象", isinstance(mylist_iter, Iterable))
print("mylist_iter是否是迭代器", isinstance(mylist_iter, Iterator))

运行效果：

4.4 案例

class MyList(object):
    """自定义的一个可迭代对象"""
    def __init__(self):
        self.items = []

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

    def __iter__(self):
        myiterator = MyIterator(self)
        return myiterator


class MyIterator():
    """
    自定义共上面可迭代对象使用的一个迭代器
    """
    def __init__(self, mylist):
        self.mylist = mylist
        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


if __name__ == '__main__':
    mylist = MyList()
    mylist.add(1)
    mylist.add(2)
    mylist.add(3)
    mylist.add(4)
    mylist.add(5)
    for num in mylist:
        print(num)

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

使用迭代器的方式实现一个不确定个数的斐波那契数列

import time


class FibIterator:
    """
    斐波那契数列迭代器
    """

    def __init__(self):
        # num1 用来保存前前一个数，初始值为数列中的第一个数1
        self.num1 = 1
        # num2 用来保存前一个数，初始值为数列中的第二个数1
        self.num2 = 1

    def __next__(self):
        """
        被next（）函数调用来获取下一个数
        :return:
        """
        temp_num = self.num1
        self.num1, self.num2 = self.num2, self.num1 + self.num2
        return temp_num
    
    def __iter__(self):
        return self
    
    
fib = FibIterator()

for i in fib:
    print(i)
    
    time.sleep(1)

效果：

总结

• 凡是可作用与for循环的对象都是Iterable对象
• 凡是课作用于next（）函数的对象都是iterator对象
• 集合数据类型如list 、dict、str等是 Iterable。但不是Iterator，不过可以通过 iter()函数获得一个 Iterator对象