个人主页:丷从心·
系列专栏:Python基础
学习指南:Python学习指南
什么是迭代
- 迭代是访问序列类型元素的一种方式
nums = [1, 2, 3, 4, 5, 6]
for num in nums:
print(num)
1
2
3
4
5
6
- 可以使用
for
循环对string
、list
、tuple
、dict
、set
等序列数据类型进行遍历,从中依次获取元素,这个过程称为迭代
可迭代对象
- 是否所有的数据类型都可以通过
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
- 可以看到
string
、list
、tuple
、dict
、set
数据类型是Iterable
类的实例,是可迭代类型 int
、float
数据类型不是Iterable
类的实例,不是可迭代类型
迭代器
- 迭代器是一个记录遍历位置的对象
- 迭代器对象从第一个元素开始访问,直至对所有的元素进行了访问
- 迭代器只能前进,不能后退
对可迭代对象进行迭代的本质
-
在对可迭代对象进行迭代的过程中,每迭代一步都会返回对象的下一个元素数据,一直向后读取元素数据,直至迭代了所有的元素
-
在这个过程中应该存在记录当前遍历位置的对象,以便每一步迭代都能返回下一条数据,这个对象就是迭代器
-
可迭代对象的本质就是提供了进行数据迭代的迭代器对象的对象类型
-
list
、tuple
等都是可迭代对象,可以通过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 4次next()
显然无法获取到第 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()
,返回包含所有数据的列表