Python的迭代器和生成器使用实例


Posted in Python onJanuary 14, 2015

一、迭代器Iterators

迭代器仅是一容器对象,它实现了迭代器协议。它有两个基本方法:

1)next方法
返回容器的下一个元素

2)__iter__方法
返回迭代器自身

迭代器可使用内建的iter方法创建,见例子:

>>> i = iter('abc')

>>> i.next()

'a'

>>> i.next()

'b'

>>> i.next()

'c'

>>> i.next()

Traceback (most recent call last):

  File "<string>", line 1, in <string>

StopIteration:
class MyIterator(object):

  def __init__(self, step):

  self.step = step

  def next(self):

  """Returns the next element."""

  if self.step==0:

  raise StopIteration

  self.step-=1

  return self.step

  def __iter__(self):

  """Returns the iterator itself."""

  return self

for el in MyIterator(4):

  print el

--------------------

结果:
3

2

1

0

二、生成器Generators

从Python2.2起,生成器提供了一种简洁的方式帮助返回列表元素的函数来完成简单和有效的代码。
它基于yield指令,允许停止函数并立即返回结果。

此函数保存其执行上下文,如果需要,可立即继续执行。

例如Fibonacci函数:

def fibonacci():

  a,b=0,1

  while True:

  yield b

  a,b = b, a+b

fib=fibonacci()

print fib.next()

print fib.next()

print fib.next()

print [fib.next() for i in range(10)]

--------------------

结果:
1

1

2

[3, 5, 8, 13, 21, 34, 55, 89, 144, 233]

PEP Python Enhancement Proposal Python增强建议

tokenize模块

>>> import tokenize

>>> reader = open('c:/temp/py1.py').next

>>> tokens=tokenize.generate_tokens(reader)

>>> tokens.next()

(1, 'class', (1, 0), (1, 5), 'class MyIterator(object):/n')

>>> tokens.next()

(1, 'MyIterator', (1, 6), (1, 16), 'class MyIterator(object):/n')

>>> tokens.next()

(51, '(', (1, 16), (1, 17), 'class MyIterator(object):/n')

例子:
def power(values):

  for value in values:

  print 'powering %s' %value

  yield value

def adder(values):

  for value in values:

  print 'adding to %s' %value

  if value%2==0:

  yield value+3

  else:

  yield value+2

elements = [1,4,7,9,12,19]

res = adder(power(elements))

print res.next()

print res.next()

--------------------

结果:
powering 1

adding to 1

3

powering 4

adding to 4

7

保持代码简单,而不是数据。
注意:宁可有大量简单的可迭代函数,也不要一个复杂的一次只计算出一个值的函数。

例子:

def psychologist():

  print 'Please tell me your problems'

  while True:

  answer = (yield)

  if answer is not None:

  if answer.endswith('?'):

  print ("Don't ask yourself too much questions")

  elif 'good' in answer:

  print "A that's good, go on"

  elif 'bad' in answer:

  print "Don't be so negative"

free = psychologist()

print free.next()

print free.send('I feel bad')

print free.send("Why I shouldn't ?")

print free.send("ok then i should find what is good for me")

--------------------

结果:
Please tell me your problems

None

Don't be so negative

None

Don't ask yourself too much questions

None

A that's good, go on

None
Python 相关文章推荐
python通过zlib实现压缩与解压字符串的方法
Nov 19 Python
NumPy 如何生成多维数组的方法
Feb 05 Python
python通过zabbix api获取主机
Sep 17 Python
python 判断字符串中是否含有汉字或非汉字的实例
Jul 15 Python
django之状态保持-使用redis存储session的例子
Jul 28 Python
Python中字典与恒等运算符的用法分析
Aug 22 Python
python 字符串常用方法汇总详解
Sep 16 Python
使用Python给头像戴上圣诞帽的图像操作过程解析
Sep 20 Python
Pytorch中的variable, tensor与numpy相互转化的方法
Oct 10 Python
Python数据相关系数矩阵和热力图轻松实现教程
Jun 16 Python
python 实现&quot;神经衰弱&quot;翻牌游戏
Nov 09 Python
如何在scrapy中集成selenium爬取网页的方法
Nov 18 Python
python实现带验证码网站的自动登陆实现代码
Jan 12 #Python
Python三元运算实现方法
Jan 12 #Python
Python中的True,False条件判断实例分析
Jan 12 #Python
Python基类函数的重载与调用实例分析
Jan 12 #Python
Python类的专用方法实例分析
Jan 09 #Python
Python序列之list和tuple常用方法以及注意事项
Jan 09 #Python
Python中的生成器和yield详细介绍
Jan 09 #Python
You might like
php处理restful请求的路由类分享
2014/02/27 PHP
PHP函数extension_loaded()用法实例
2015/01/19 PHP
PHP遍历XML文档所有节点的方法
2015/03/12 PHP
培养自己的php编码规范
2015/09/28 PHP
PHP实现计算器小功能
2020/08/28 PHP
Javascript select下拉框操作常用方法
2009/11/09 Javascript
Jquery.TreeView结合ASP.Net和数据库生成菜单导航条
2010/08/27 Javascript
在IE浏览器中resize事件执行多次的解决方法
2011/07/12 Javascript
jQuery动态地获取系统时间实现代码
2013/05/24 Javascript
JavaScript Sort 的一个错误用法示例
2015/03/20 Javascript
js简单实现竖向tab选项卡的方法
2015/05/04 Javascript
JavaScript数组迭代器实例分析
2015/06/09 Javascript
javascript背景时钟实现方法
2015/06/18 Javascript
使用OpenLayers3 添加地图鼠标右键菜单
2015/12/29 Javascript
JS中的eval 为什么加括号
2016/04/13 Javascript
javascript运算符语法全面概述
2016/07/14 Javascript
Highcharts学习之数据列
2016/08/03 Javascript
jquery ajaxfileupload异步上传插件使用详解
2017/02/08 Javascript
Flutter部件内部状态管理小结之实现Vue的v-model功能
2019/06/11 Javascript
jQuery实现王者荣耀手风琴效果
2020/01/17 jQuery
python制作爬虫爬取京东商品评论教程
2016/12/16 Python
pygame实现弹力球及其变速效果
2017/07/03 Python
Python调用C# Com dll组件实战教程
2017/10/12 Python
Pytorch 定义MyDatasets实现多通道分别输入不同数据方式
2020/01/15 Python
深入了解NumPy 高级索引
2020/07/24 Python
Python如何读取、写入JSON数据
2020/07/28 Python
Python Celery异步任务队列使用方法解析
2020/08/10 Python
运动会通讯稿300字
2014/02/02 职场文书
企业文化建设实施方案
2014/03/22 职场文书
2014年教师业务学习材料
2014/05/12 职场文书
市场总监岗位职责
2015/02/11 职场文书
关于清明节的演讲稿2015
2015/03/18 职场文书
教师岗位职责范本
2015/04/02 职场文书
冰雪公主观后感
2015/06/16 职场文书
女方家长婚礼答谢词
2015/09/29 职场文书
MySQL数据库10秒内插入百万条数据的实现
2021/11/01 MySQL