Python 3 iter函数用法简述

时间:2023-11-30 11:51:02

Python 3中关于iter(object[, sentinel)]方法有两个参数。

使用iter(object)这种形式比较常见。

iter(object, sentinel)这种形式一般较少使用,Python的文档说明貌似也不容易理解。

故,在此举例说明一下此函数的用法。

1,iter(object)

Python官方文档对于这种形式的解释很容易理解。

此时,object必须是集合对象,且支持迭代协议(iteration protocol)或者支持序列协议(sequence protocol)。

说白了,也就是实现了__iter__()方法或者__getitem__()方法。

 lst = [1, 2, 3]
for i in iter(lst):
print(i)

执行结果:

1
2
3

2,iter(object, sentinel)

Python官方文档对于这种形式的解释是:“ If the second argument, sentinel, is given, then object must be a callable object. The iterator created in this case will call object with no arguments for each call to its __next__() method; if the value returned is equal to sentinel,StopIteration will be raised, otherwise the value will be returned.”。

这句话的意思是说:如果传递了第二个参数,则object必须是一个可调用的对象(如,函数)。此时,iter创建了一个迭代器对象,每次调用这个迭代器对象的__next__()方法时,都会调用object。

如果__next__的返回值等于sentinel,则抛出StopIteration异常,否则返回下一个值。

 class counter:

     def __init__(self, _start, _end):
self.start = _start
self.end = _end def get_next(self):
s = self.start
if(self.start < self.end):
self.start += 1
else:
raise StopIteration return s c = counter(1, 5)
iterator = iter(c.get_next, 3)
print(type(iterator))
for i in iterator:
print(i)

执行结果:

<class 'callable_iterator'>
1
2