机器学习之python入门指南（七）迭代器和生成器的使用

Python3迭代器和生成器的使用

1）迭代器

迭代是访问集合元素的一种方式，迭代器是可以记住遍历位置的对象，迭代器对象从集合的第一个元素开始访问，直到所有元素被访问完结束，迭代器只能往前，不会后退。

迭代器的两个方法：iter()和next()。

字符串，列表，元组对象都可以创建迭代器

运行结果为：
1
2

迭代器对象可以用于for语句进行遍历：如

 

运行结果为：1 2 3 4

创建一个迭代器。

  把一个类作为一个迭代器使用需要实现两个方法。__iter__()和__next__()

  __iter__()方法返回一个特殊的迭代器对象，这个迭代对象实现了__next__()方法并通过StopIteration异常标识迭代的完成，__next__()方法会返回一个迭代器对象。

运行结果为：
1

2

3

4

StopIteration   

  StopIteration异常用于标识迭代的完成，防止出现无限循环的情况，在 __next__() 方法中我们可以设置在完成指定循环次数后触发StopIteration 异常来结束迭代。

 
运行结果为：

1 

2 

3 

4

5 

6

 

生成器

  在Python中，使用了yield的函数被称为生成器，生成器是一个返回迭代器的函数，只能用于迭代操作，更简单点理解就是生成器就是一个迭代器。

  在调用生成器的过程中，每次遇到yield函数时，函数会暂停并保存当前所有的运行信息，返回yield的值，并在下一次执行next()方法时从当前位置继续运行。调用一个生成器，返回的是一个迭代器对象。