可迭代对象和迭代器

Posted 2021-02-20 my_captain

可迭代对象(iterable)

定义

实现了__iter__方法的对象称为可迭代对象，在python中，可迭代对象有字符串、列表、元组、字典、集合等

创建可迭代对象

class MyNumbers:

    def __init__(self):
        self.a = 1

    def __iter__(self):
        return self

my_numbers = MyNumbers()

判断

判断对象是否是可迭代对象

from collections.abc import Iterable

print(isinstance("123", Iterable))        #字符串：True
print(isinstance([1, 2, 3], Iterable))    #列表：True
print(isinstance((1, 2, 3), Iterable))    #元组：True
print(isinstance({"a": 1, "b": 2}, Iterable))   #字典：True
print(isinstance({1, 2, 3}, Iterable))    #集合：True
print(isinstance(open("test.txt", "r"), Iterable))  #文件：True
print(isinstance(my_numbers, "r"), Iterable))   #my_numbers对象：True

也可以去源码里查看，比如源码里的list类实现了__iter__方法

迭代器(iterator)

定义

实现了__iter__和__next__方法的对象称为迭代器，__iter__返回一个特殊的迭代器对象self，__next__返回下一个可用元素，如果没有元素，会抛出StopIteration异常

内置的迭代器

map()、zip()、enumerate()、filter()、文件对象等都是迭代器

创建迭代器

class MyNumbers:

    def __init__(self):
        self.a = 1

    def __iter__(self):
        return self

    def __next__(self):
        x = self.a
        self.a += 1
        return x

my_numbers = MyNumbers()
print(next(my_numbers))  #1
print(next(my_numbers))  #2
print(next(my_numbers))  #3

判断

from collections.abc import Iterator

print(isinstance(my_numbers, Iterator))  #my_numbers对象：True

特点

• 迭代器是一个可以记住遍历位置的对象
• 迭代器对象从集合的第一个元素可以访问，直到所有的元素被访问完结束
• 迭代器只能往前不能后退
• 迭代器有两个基本的方法：iter()和next()

可迭代对象和迭代器的关系

用可迭代对象创建迭代器

可迭代对象实现了__iter__方法，但没有实现__next__方法，实际上可迭代对象调用iter()方法时，它会自动调用__iter__方法，而__iter__方法的作用是返回一个特殊的迭代器，因此实现了用可迭代对象创建迭代器

用列表创建迭代器

L = [1, 2, 3]
I = iter(L)
print(next(I))
print(next(I))
print(next(I))
print(next(I))  #没有下一个元素，抛出异常
print(next(I))


# 运行结果
Traceback (most recent call last):
  File "E:/virtual_workshop/everyday/test/iterators.py", line 80, in <module>
    print(next(I))
StopIteration
1
2
3

为了避免异常的情况，可以使用for或者next加异常捕获来遍历元素
for循环

L = [1, 2, 3]
I = iter(L)

for i in I:
    print(i, end=" ")


# 运行结果
1 2 3

next加异常捕获

import sys

L = [1, 2, 3]
I = iter(L)

while True:
    try:
        print(next(I), end=" ")
    except StopIteration:
        sys.exit()


# 运行结果
1 2 3

用字典创建迭代器

D = {"a":1, "b": 2, "c": 3}
I = iter(D)

for i in I:
    print(i, end=" ")


# 运行结果
a b c

字典中有一个迭代器（但不能说字典本身就是迭代器），在迭代环境中，会自动一次返回一个key

D = {"a":1, "b": 2, "c": 3}

for key in D:
    print(key, end=" )


# 运行结果
a b c

关系

• 迭代器一定是可迭代的，但可迭代对象不是迭代器
• 可以用可迭代对象创建迭代器
• 可迭代对象必须实现__iter__方法，不能实现__next__方法
• 可迭代对象与迭代器都有__iter__方法，但可迭代对象的__iter__用于实例化一个迭代器对象，而迭代器中的__iter__用于返回迭代器本身，与__next__共同完成迭代器的迭代

参考文章

《Python3 迭代器与生成器》
《python3 迭代器》
《Python3学习笔记（十四）：可迭代对象、迭代器和生成器》
《如何理解Python中的可迭代对象、迭代器和生成器》