如何实现可迭代的对象和迭代器

iter(...)
iter(iterable) -> iterator
iter(callable, sentinel) -> iterator

Get an iterator from an object.  In the first form, the argument must
supply its own iterator, or be a sequence.
In the second form, the callable is called until it returns the sentinel.

可以由可迭代对象得到一个迭代器

l = [1,2,3,4]
print(iter(l)) # <list_iterator object at 0x000002D7D2D7B208>
print(l.__iter__()) # <list_iterator object at 0x000001E6BE719198>

可迭代对象满足了迭代协议的一个接口 iter 或者 getitem
迭代器对象只有一个接口next
next 可以得到迭代器中的下一个元素,最后一个元素被取出后,会抛出StopIteration异常

for循环工作机制:
1.由iter方法得到一个迭代器
2.不停地调用next方法,直到捕获一个会抛出StopIteration异常,退出循环

如何实现可迭代对象和迭代器对象

实际案例:
某软件要求,从网络抓取各个城市的气温信息,并以此显示:
北京:15-20
天津:17-22
长春:12-18
...

如果一次抓取所有的城市天气再显示,显示第一个城市气温时,有很高的延时,并且浪费存储空间。
我们期望以“用时访问”策略,并且能把所有的城市气温封装到一个对象里,可用for语句进行迭代,如何解决?

解决方案:

  1. 实现一个迭代器对象WeatherIterator,next方法每次返回一个气温
  2. 实现一个可迭代对象WeatherIterable,iter方法返回一个迭代器对象
import requests
from collections import Iterable,Iterator

class WeatherIterator(Iterator):
    """
    迭代器对象,满足__next__接口
    """
    def __init__(self,citys):
        self.citys = citys
        self.index = 0

    def _getCityWeather(self,city):
        r = requests.get("http://wthrcdn.etouch.cn/weather_mini?city=%s" % city)
        data = r.json()['data']['forecast'][0]
        return "%s:%s-%s" % (city, data['low'], data['high'])

    def __next__(self):
        if self.index == len(self.citys):
            raise StopIteration
        city = self.citys[self.index]
        self.index += 1
        return self._getCityWeather(city)

class WeatherIterable(Iterable):
    """
    可迭代对象,满足__iter__接口
    """
    def __init__(self,citys):
        self.citys = citys

    def __iter__(self):
        return WeatherIterator(self.citys)

citys = ['北京', '上海', '广州', '深圳']
weather_obj = WeatherIterable(citys)

for x in weather_obj:
    print(x)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • PYTHON-进阶-ITERTOOLS模块小结转自wklken:http://wklken.me/posts/20...
    C_Y_阅读 1,047评论 0 2
  • 迭代器(Iterable) 简单来说,迭代器对象(my_list)可以让以下代码正常工作: 如果对象实现了__it...
    SOLAREST阅读 574评论 0 1
  • 1.1==,is的使用 ·is是比较两个引用是否指向了同一个对象(引用比较)。 ·==是比较两个对象是否相等。 1...
    TENG书阅读 746评论 0 0
  • Python中,所有的集合都可以迭代。在语言内部中,迭代器主要用于: for循环 构建和扩展集合类型 逐行遍历文件...
    F4A3阅读 370评论 0 0
  • 世界上本没有路 走的人多了 就变成了路 道路本是有始有终 可怕的不是中途迷路 而是中途失去了方向 前一段时间 表妹...
    静脉武士阅读 177评论 0 0