Python函数进阶之迭代器的原理与使用详解

# 定义一个列表，列表是可迭代对象
lst = [1, 2, 3, 4, 5]

# 获取列表的所有成员
res_lst = dir(lst)

print(res_lst)
'''
['__add__', '__class__', '__contains__', '__delattr__', '__delitem__', '__dir__', '__doc__', '__eq__', '__format__', '__ge__', '__getattribute__', '__getitem__', '__gt__', '__hash__', '__iadd__', '__imul__', '__init__', '__init_subclass__', '__iter__', '__le__', '__len__', '__lt__', '__mul__', '__ne__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__reversed__', '__rmul__', '__setattr__', '__setitem__', '__sizeof__', '__str__', '__subclasshook__', 'append', 'clear', 'copy', 'count', 'extend', 'index', 'insert', 'pop', 'remove', 'reverse', 'sort']
'''

# 查看是否存在__iter__方法
res = '__iter__' in res_lst

print(res)  # True

# 存在__iter__方法，说明确实是一个可迭代对象

lst = [1, 2, 3]
print(type(lst))    # <class 'list'>
it = iter(lst)
print(type(it))     # <class 'list_iterator'>

lst = [1, 2, 3]
print(type(lst))    # <class 'list'>
it = lst.__iter__()
print(type(it))     # <class 'list_iterator'>

# 列表
lst = list()

# 迭代器
lst_it = iter(lst)

# 迭代器中的所有成员
res_lst = dir(lst_it)

# 判断
if '__iter__' in res_lst:
	print('lst_it是一个可迭代对象')

if '__next__' in res_lst:
	print('lst_it是一个迭代器')

'''
结果：
lst_it是一个可迭代对象
lst_it是一个迭代器
'''

lst = list()

lst_it = iter(lst)

# 判断是否是迭代器
res = isinstance(lst_it, Iterator)
print(res)  # True

# 判断是否是可迭代对象
res = isinstance(lst_it, Iterable)
print(res)  # True

# 因为lst本没有数据，所以无法取出数据
lst = list()
lst_it = iter(lst)
res = next(lst_it)  # StopIteration
print(res)

lst = [1, 2, 3]

lst_it = iter(lst)

# 迭代器中一次只会取出一个数据
print(next(lst_it))  # 1
print(next(lst_it))  # 2
print(next(lst_it))  # 3

# 超出迭代器中的元素个数，就会报错
print(next(lst_it))  # StopIteration

# 如果要重新取出数据，就重置迭代器，重新定义一边迭代器就是重置迭代器
lst_it = iter(lst)

# 再次取出数据，使用__next__方法
print(lst_it.__iter__())  # 1
print(lst_it.__iter__())  # 2
print(lst_it.__iter__())  # 3