【Python基础】迭代器(6722字)

本文阐述了Python中的迭代机制,包括判断数据类型是否可迭代,迭代器的使用,自定义迭代器示例,for循环底层原理,以及多种接收可迭代对象的方法。
摘要由CSDN通过智能技术生成

因上努力

个人主页:丷从心·

系列专栏:Python基础

学习指南:Python学习指南

果上随缘


什么是迭代

  • 迭代是访问序列类型元素的一种方式
nums = [1, 2, 3, 4, 5, 6]

for num in nums:
    print(num)
1
2
3
4
5
6
  • 可以使用for循环对stringlisttupledictset等序列数据类型进行遍历,从中依次获取元素,这个过程称为迭代

可迭代对象

  • 是否所有的数据类型都可以通过for循环进行迭代呢?
tel = 10086

for item in tel:
    print(item)
Traceback (most recent call last):
  File "C:/Users/FOLLOW_MY_HEART/Desktop/Python Basics/【Python基础】迭代器/test.py", line 3, in <module>
    for item in tel:
TypeError: 'int' object is not iterable
  • 通过运行发现TypeError异常,显示int类型不是iterable类型,即可迭代类型
判断数据类型是否为可迭代类型
from collections.abc import Iterable

# 判断 string 类型是否为可迭代类型
print(isinstance('', Iterable))
# 判断 list 类型是否为可迭代类型
print(isinstance([], Iterable))
# 判断 tuple 类型是否为可迭代类型
print(isinstance((), Iterable))
# 判断 dict 类型是否为可迭代类型
print(isinstance({'': ''}, Iterable))
# 判断 set 类型是否为可迭代类型
print(isinstance({}, Iterable))

# 判断 int 类型是否为可迭代类型
print(isinstance(0, Iterable))
# 判断 float 类型是否为可迭代类型
print(isinstance(3.14, Iterable))
True
True
True
True
True
False
False
  • 可以看到stringlisttupledictset数据类型是Iterable类的实例,是可迭代类型
  • intfloat数据类型不是Iterable类的实例,不是可迭代类型

迭代器

  • 迭代器是一个记录遍历位置的对象
  • 迭代器对象从第一个元素开始访问,直至对所有的元素进行了访问
  • 迭代器只能前进,不能后退
对可迭代对象进行迭代的本质
  • 在对可迭代对象进行迭代的过程中,每迭代一步都会返回对象的下一个元素数据,一直向后读取元素数据,直至迭代了所有的元素

  • 在这个过程中应该存在记录当前遍历位置的对象,以便每一步迭代都能返回下一条数据,这个对象就是迭代器

  • 可迭代对象的本质就是提供了进行数据迭代的迭代器对象的对象类型

  • listtuple等都是可迭代对象,可以通过iter()函数获取这些可迭代对象提供的迭代器,然后可以通过对迭代器使用next()函数来获取下一条数据

获取可迭代对象的迭代器
  • 通过iter()函数获取可迭代对象的迭代器
from collections.abc import Iterator

number_list = [1, 2, 3, 4, 5, 6]

number_list_iter = iter(number_list)  # 获取 list 对象的迭代器

print(type(number_list_iter))

print(isinstance(number_list, Iterator))
print(isinstance(number_list_iter, Iterator))
<class 'list_iterator'>
False
True
通过迭代器获取数据
  • 通过对迭代器使用next()函数来获取下一条数据
number_list = [1, 2, 3]

number_list_iter = iter(number_list)  # 获取 list 对象的迭代器

number_1 = next(number_list_iter)
print(number_1)

number_2 = next(number_list_iter)
print(number_2)

number_3 = next(number_list_iter)
print(number_3)
1
2
3
StopIteration异常
  • 如果将上面的代码多调用一次next()会发生什么?
number_list = [1, 2, 3]

number_list_iter = iter(number_list)  # 获取 list 对象的迭代器

number_1 = next(number_list_iter)
print(number_1)

number_2 = next(number_list_iter)
print(number_2)

number_3 = next(number_list_iter)
print(number_3)

number_4 = next(number_list_iter)  # 这里会产生异常
print(number_4)
1
2
3
Traceback (most recent call last):
  File "C:/Users/FOLLOW_MY_HEART/Desktop/Python Basics/【Python基础】迭代器/test.py", line 14, in <module>
    number_4 = next(number_list_iter)  # 这里会产生异常
StopIteration
  • 可以看到第 14 14 14行代码处产生了异常
  • 因为列表nums中只有 3 3 3个元素,调用 4 4 4next()显然无法获取到第 4 4 4个元素
  • 所以StopIteration异常其实是一种告知迭代结束的标志而已
  • 添加异常处理即可解决问题
number_list = [1, 2, 3]

number_list_iter = iter(number_list)  # 获取 list 对象的迭代器

number_1 = next(number_list_iter)
print(number_1)

number_2 = next(number_list_iter)
print(number_2)

number_3 = next(number_list_iter)
print(number_3)

try:
    number_4 = next(number_list_iter)  # 这里会产生异常
    print(number_4)
except StopIteration as e:
    print('迭代结束...')
1
2
3
迭代结束...

自定义迭代器

  • 上面提到的iter()方法必须是从可迭代对象提取迭代器对象,但是怎样保证自定义对象是可迭代对象呢?
_iter_()方法
无_iter_()方法
from collections.abc import Iterable


class MyList:
    def __init__(self):
        self.my_list = list()

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


my_list = MyList()

print(isinstance(my_list, Iterable))
False
有_iter_()方法
from collections.abc import Iterable


class MyList:
    def __init__(self):
        self.my_list = list()

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

    def __iter__(self):
        pass


my_list = MyList()

print(isinstance(my_list, Iterable))
True
  • 可以看出,只要一个类定义了__iter__()方法,那么这个类的实例就是可迭代对象
  • 实际上,当调用iter()方法获取一个可迭代对象的迭代器时,会自动触发这个对象的__iter__()方法,返回这个对象的迭代器
_next_()方法
  • 通过对迭代器使用next()函数能够获取下一条数据,实际上会自动触发这个对象的__next__()方法
  • 所以想要构造一个迭代器,就要实现__next__()方法
  • Python要求迭代器本身也是可迭代的,所以还要为迭代器类实现__iter__()方法,而__iter__()方法要返回一个迭代器,迭代器自身正是一个迭代器,所以迭代器类的__iter__()方法返回自身即可
  • 一个实现了__iter__()方法和__next__()方法的对象就是迭代器
判断数据类型是否为迭代器类型
  • 使用isinstance()判断一个对象是否是Iterator对象
from collections.abc import Iterable, Iterator

number_list = [1, 2, 3]

print(isinstance(number_list, Iterable))
print(isinstance(number_list, Iterator))

number_list_iter = iter(number_list)

print(isinstance(number_list_iter, Iterable))
print(isinstance(number_list_iter, Iterator))
True
False
True
True
自定义迭代器案例
分离构造
class MyList:
    def __init__(self):
        self.items = list()
        self.current = 0  # 记录当前迭代位置

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

    def __iter__(self):
        return MyIterator(self)  # 返回一个迭代器对象


class MyIterator:
    def __init__(self, my_list_obj):
        self.my_list_obj = my_list_obj

    def __iter__(self):
        return self

    def __next__(self):
        if self.my_list_obj.current < len(self.my_list_obj.items):
            item = self.my_list_obj.items[self.my_list_obj.current]

            self.my_list_obj.current += 1  # 当前迭代位置加 1

            return item
        else:
            raise StopIteration  # 手动抛出异常


if __name__ == '__main__':
    my_list = MyList()

    my_list.add(1)
    my_list.add(2)
    my_list.add(3)

    iter_obj = iter(my_list)

    print(next(iter_obj))  # 打印 1

    # 打印 2、3, 思考一下为什么 for 循环从 2 开始打印
    for item in my_list:
        print(item)
1
2
3
整合构造
class MyList:
    def __init__(self):
        self.items = list()
        self.current = 0  # 记录当前迭代位置

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

    def __iter__(self):
        return self  # 返回一个迭代器对象

    def __next__(self):
        if self.current < len(self.items):
            item = self.items[self.current]

            self.current += 1  # 当前迭代位置加 1

            return item
        else:
            raise StopIteration  # 手动抛出异常


if __name__ == '__main__':
    my_list = MyList()

    my_list.add(1)
    my_list.add(2)
    my_list.add(3)

    iter_obj = iter(my_list)

    print(next(iter_obj))  # 打印 1

    # 打印 2、3, 思考一下为什么 for 循环从 2 开始打印
    for item in my_list:
        print(item)
1
2
3
for循环的本质
number_list = [1, 2, 3, 4, 5, 6]

number_list_iter = iter(number_list)
while True:
    try:
        print(next(number_list_iter))
    except StopIteration:
        break
1
2
3
4
5
6
  • 在对可迭代对象进行迭代时,for循环先调用iter()方法,会自动触发这个对象的__iter__()方法,返回这个对象的迭代器
  • 然后,对获取到的迭代器不断调用next()方法,会自动触发这个对象的__next__()方法,返回下一条数据
  • 最后,通过处理StopIteration异常来结束循环

接收可迭代对象的方式

  • 并不是只有for循环能够接收可迭代对象
  • list()tuple()等也能接收可迭代对象
示例
class MyList:
    def __init__(self):
        self.items = list()
        self.current = 0  # 记录当前迭代位置

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

    def __iter__(self):
        return self  # 返回一个迭代器对象

    def __next__(self):
        if self.current < len(self.items):
            item = self.items[self.current]

            self.current += 1  # 当前迭代位置加 1

            return item
        else:
            raise StopIteration  # 手动抛出异常


if __name__ == '__main__':
    my_list = MyList()

    my_list.add(1)
    my_list.add(2)
    my_list.add(3)

    iter_obj = iter(my_list)

    print(list(iter_obj))
[1, 2, 3]
  • 对迭代器iter_obj调用list(),返回包含所有数据的列表

  • 24
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值