Python 3 iter函数用法简述
Python 3中关于iter(object[, sentinel)]方法有两个参数。
使用iter(object)这种形式比较常见。
iter(object, sentinel)这种形式一般较少使用,Python的文档说明貌似也不容易理解。
故,在此举例说明一下此函数的用法。
1,iter(object)
Python官方文档对于这种形式的解释很容易理解。
此时,object必须是集合对象,且支持迭代协议(iteration protocol)或者支持序列协议(sequence protocol)。
说白了,也就是实现了__iter__()方法或者__getitem__()方法。
1 lst = [1, 2, 3]
2 for i in iter(lst):
3 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异常,否则返回下一个值。
1 class counter:
2
3 def __init__(self, _start, _end):
4 self.start = _start
5 self.end = _end
6
7 def get_next(self):
8 s = self.start
9 if(self.start < self.end):
10 self.start += 1
11 else:
12 raise StopIteration
13
14 return s
15
16
17 c = counter(1, 5)
18 iterator = iter(c.get_next, 3)
19 print(type(iterator))
20 for i in iterator:
21 print(i)
执行结果:
<class 'callable_iterator'>
1
2