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

posted @ 2016-03-04 19:10  腾鲲  阅读(25460)  评论(0编辑  收藏  举报