本节简单介绍下Python 中如何创建迭代器及如何使用,后续如果有其他注意点持续更新
先上示例代码,注释已经大概介绍了迭代器的相关信息
本节用到了收集参数(*args, **kwargs),如果不了解可以网上搜索下
# 同时拥有__iter__及__next__方法的对象就是迭代器
class A:
def __init__(self):
self.number = 0
# __iter__用于生成迭代器
def __iter__(self):
return self
# __next__用于获取迭代器下一个元素
def __next__(self):
if self.number < 30:
self.number += 1
return self.number
# 设置迭代范围,当number超出30时触发异常
else:
raise StopIteration
# 应用于map的函数
def add_num(int1):
return int1 + 5
try:
insA1 = A()
# list()用于将迭代器值全部列出
# print('list of insA1: {}'.format(list(insA1)))
# map()用于将迭代器的每个值应用于函数并返回一个迭代器,通过next()函数可获取map迭代器的值
print(next(map(add_num, insA1)))
# iter()用于基于对象生成迭代器,但对象必须要遵守迭代器协议
print(iter(insA1))
print(next(insA1))
# 利用for循环可以循环调用next获取迭代器数值
for iNum in range(10):
print('next of iter is {}'.format(next(insA1)))
# 更简单的,for循环语句可以直接对迭代器对象进行迭代
for iNum in insA1:
if iNum < 20:
print('Found value below 20 is {}'.format(iNum))
except StopIteration:
print('Function exit')
---
总结
注意几点:
1、 迭代器协议必须要有__iter__及__next__魔法方法,这是必要的
2、迭代器的作用主要是在处理大数据时可以保证较小的内存及算力占用,在查找数据时只会去找到下一个而不是一次性全部查找可以平衡计算能力,在实际中的应用举例,要在一个城市中查找年龄在18以上的人,迭代器的方法就是找到一个人就给你返回其信息,如果简单的用For语句时就会将城市中所有人找到然后再给你返回信息,但这样会相当消耗时间对于程序而言就是可能会宕机一段时间,因为语句没有执行完成