Python 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