迭代器必须满足两个条件,一个是可迭代,一个是可以让next()调用,例如列表,元组,字符串等都是可以迭代的,但无法让next()调用,因此不属于迭代器,迭代器的主要好处在于节省空间。
例如[1,2,3,4]可以调任意值,原因是内存中开辟了四块空间,如果是迭代器的话,一般是从头元素开始,如果得到下一个元素,通过某种计算可以得到下个元素,第一个元素则消失不见,所以总的来说只开辟了一块空间,从而节省了空间,但会影响性能。
list=[1,2,3,4]
it = iter(list) # 可以将可迭代的对象转化成迭代器,但一般不会这么做,因为list已经占了一部空间,再使用迭代方法也会占用空间,所以这并没有节省空间
print(next(it))
print(next(it))
print(next(it))
print(next(it))
print(next(it))
返回结果:
print(next(it))StopIteration
1
2
3
4
当元素迭代结束后,超出会报一个StopIteration异常
创建一个迭代器
# 迭代器必须包含__iter__,__next__, 主要的计算在next下面
# 案例:输入n,返回1-n的平方,
class myiter():
def __init__(self,n):
self.n = n
self.first = 1
def __iter__(self):
return self # 固定语法
def __next__(self):
if self.first>self.n:
raise StopIteration # 当迭代器遇到该异常时,则结束
data = self.first**2
self.first+=1
return data
it = myiter(2)
for i in it:
print(i)
返回结果:
1
4