23. 迭代器

一、什么是迭代器

迭代器 指的是迭代取值的工具，迭代是一个重复的过程，每次重复都是基于上一次的结果而继续的，单纯的重复并不是重复。迭代器是用来迭代取值的工具，而涉及到把多个值循环取出的数据类型有：列表、字符串、元组、字典、集合、打开的文件对象等。

我们可以通过 while 循环的方式取值，这种取值方式只适用于有索引的数据类型，如：列表、字符串、元组等。

names = ["Sakura","Mikoto","Shana"]i = 0
while i < len(names):print(names[i])i += 1

为了解决基于索引迭代器取值的局限性，Python 需要提供一种能够不依赖索引的取值方式，这就是迭代器。在 Python 中，但凡是内置有 __iter__() 方法的对象都称为 可迭代对象。在 Python 中，列表、字符串、元组、字典、集合、打开的文件对象等都内置了 __iter__() 方法。调用可迭代对象的 __iter__() 方法会将其转换成迭代器对象。转换为迭代器对象后，我们可以通过 __next__() 方法获取迭代器对象的下一个值。

names = ["Sakura", "Mikoto", "Shana"]iterator = person.__iter__()
print(iterator)
print(type(iterator), end="\n\n")while True:try:print(iterator.__next__())except StopIteration:break

names = ["Sakura", "Mikoto", "Shana"]iterator = iter(names)
print(iterator)
print(type(iterator), end="\n\n")while True:try:print(next(iterator))except StopIteration:break

如果我们迭代到最后一项，再调用 __next__() 方法或 next() 方法会抛出 StopIteration 异常；

迭代器对象 是指内置有 __next__() 方法和 __iter__() 方法的对象。调用迭代器对象的 __next__() 方法将得到迭代器对象的下一个值。调用迭代器对象的 __iter__() 方法将得到迭代器的本身，与没调用该方法一样。

person = {"name":"Sakura","age":10}iterator = iter(person)
print(iterator is  iterator.__iter__() is iterator.__iter__().__iter__())

如果我们想判断某一个数据类型或对象是否可以迭代，我们可以使用 collections 模块来判断；

from collections.abc import Iterable, Iteratordata = "Sakura"
print("可迭代对象-字符串：", isinstance(data, Iterable))
print("迭代器-字符串：", isinstance(data, Iterator))
print()data = ["Sakukra", "Mikoto", "Shana"]
print("可迭代对象-列表：", isinstance(data, Iterable))
print("迭代器-列表：", isinstance(data, Iterator))
print()data = ("Sakukra", "Mikoto", "Shana")
print("可迭代对象-元组：", isinstance(data, Iterable))
print("迭代器-元组：", isinstance(data, Iterator))
print()data = {"Sakukra", "Mikoto", "Shana"}
print("可迭代对象-集合：", isinstance(data, Iterable))
print("迭代器-集合：", isinstance(data, Iterator))
print()data = {"name": "Sakura", "age": 10}
print("可迭代对象-字典：", isinstance(data, Iterable))
print("迭代器-字典：", isinstance(data, Iterator))
print()

from collections.abc import Iterable, Iteratorclass People:def __init__(self):self.container = []people = People()
print("可迭代对象：", isinstance(people, Iterable))
print("迭代器：", isinstance(people, Iterator))

二、自定义迭代器

只要在类中定义 __iter__() 方法，那么这个类创建出来的对象一定是可迭代的。通俗的说：一个具备了 __iter__() 方法的对象，就是一个可迭代对象。

from collections.abc import Iterable, Iteratorclass People:def __init__(self):self.container = []def __iter__(self):passpeople = People()
print(isinstance(people, Iterable))
print("迭代器：", isinstance(people, Iterator))

如果想实现自定义迭代器的话，除了在类中定义 __iter__() 方法还需要在类中再定义 __next__() 方法，即一个实现了 __iter__() 方法和 __next__() 方法的对象，就是迭代器。

from collections.abc import Iterable, Iteratorclass People:def __init__(self):self.container = []def add(self, name):self.container.append(name)def __iter__(self):# 这个方法有两个功能：# 1.标记用当前类创建出来的对象是一定是可迭代对下个你# 2.当调用iter()函数时，这个方法会被自动调用，返回的自己指定的那个迭代器return MyIterator(self)class MyIterator:def __init__(self, obj):self.obj = objself.current = 0def __iter__(self):passdef __next__(self):# 这个方法有两个功能：# 1.标记当前类创建的对象（一定有__iter__()方法）一定是迭代器# 2.当调用next()函数时，这个方法会自动调用，它返回一个数据if self.current < len(self.obj.container):item = self.obj.container[self.current]self.current += 1return itemelse:raise StopIterationpeople = People()
print("People-可迭代对象：", isinstance(people, Iterable))
print("People-迭代器：", isinstance(people, Iterator))
print()iterator = MyIterator(people)
print("MyIterator-可迭代对象：", isinstance(iterator, Iterable))
print("MyIterator-迭代器：", isinstance(iterator, Iterator))
print()people.add("Sakura")
people.add("Mikoto")
people.add("Shana")for item in people:print(item)

for 循环是一个已经实现的功能，它里面自带了 iter()、next() 方法，并且带有 StopIteration 异常判断，通过这个异常判断来决定是否还需要获取迭代器对象。如果自己规定用 None 来表示数据获取完毕，但是 for 循环的代码依然用的是异常处理，所以 for 循环会产生死循环。

from collections.abc import Iterable, Iteratorclass People:def __init__(self):self.container = []self.current = 0def add(self, name):self.container.append(name)def __iter__(self):# 这个方法有两个功能：# 1.标记用当前类创建出来的对象是一定是可迭代对下个你# 2.当调用iter()函数时，这个方法会被自动调用，返回的自己指定的那个迭代器return selfdef __next__(self):# 这个方法有两个功能：# 1.标记当前类创建的对象（一定有__iter__()方法）一定是迭代器# 2.当调用next()函数时，这个方法会自动调用，它返回一个数据if self.current < len(self.container):item = self.container[self.current]self.current += 1return itemelse:self.current = 0raise StopIterationpeople = People()people.add("Sakura")
people.add("Mikoto")
people.add("Shana")for item in people: