Python高效编程实战---2、对象迭代与反迭代技巧

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

# -*- coding: utf-8 -*-

import requests
from collections import Iterable, Iterator

class WeatherIterator(Iterator):  # 定义迭代器 
    def __init__(self, cities):
        self.cities = cities
        self.index = 0    
        
    def get_weather(self,city):   # 获取天气的数据
        r = requests.get(u'http://wthrcdn.etouch.cn/weather_mini?city=' + city)   
        data = r.json()['data']['forecast'][0]
        return '%s:%s,%s' % (city, data['low'], data['high'])

    def next(self):     # 定义迭代器的next()方法
        if self.index == len(self.cities):
            raise StopIteration    # 如果全部迭代完的处理方法
        city = self.cities[self.index]
        self.index += 1
        return self.get_weather(city)

class WeatherIterable(Iterable):  # 定义可迭代对象
    def __init__(self, cities):
        print ' __init__'
        self.cities = cities

    def __iter__(self):
        print '__iter__'
        return WeatherIterator(self.cities)    # 返回一个迭代器对象


if __name__ == '__main__':
    # for x in obj 循环的基本原理是，
    # obj是一个可迭代对象，for循环之前，先调用obj.__iter__方法，返回一个迭代器对象Iterator，迭代器对象通过nex()t方法，返回每个值
    # 要写出自己的可迭代对象，需要继承Iterable，__iter__()方法，并返回一个可迭代器对象
    # 在自己的迭代器对象中定义next()方法，通过next方法，获取每次迭代的值

    city_list = [u'深圳', u'北京', u'上海', u'广州', u'青岛', u'荆州', u'武汉']
    for x in WeatherIterable(city_list):
        print x

二、使用生成器函数实现可迭代对象

class PrimeNumber(object):
    def __init__(self, start, end):
        self.start = start
        self.end = end

    def get_prime(self, k):  # 判断是否是素数，如果是，返回True
        if k < 2:
            return False
        for x in xrange(2, k):
            if k % x == 0:
                return False
        return True

    def __iter__(self):  # for...in..循环时，先调用__iter__，返回一个迭代生成器对象
        print '__iter__'
        for x in xrange(self.start, self.end):
            if self.get_prime(x):
 # yield是一个关键词，类似return, 不同之处在于，yield返回的是一个生成器对象，生成器每次调用next,返回一个x
                yield x 

for x in PrimeNumber(0, 100):
    print x

三、如何进行反向迭代及如何实现反向迭代

class FloatRange(object):
    def __init__(self, start, end, step=0.5):
        self.start = start
        self.end = end
        self.step = step

    def __iter__(self):   # 定义正向迭代方法
        t = self.start
        while t <= self.end:
            yield t
            t += self.step

    def __reversed__(self):   # 定义反向迭代方法
        t = self.end
        while t >= self.start:
            yield t
            t -= self.step

for x in FloatRange(1, 4, 0.5):   # 默认调用__iter__()
    print x, " ",

for x in reversed(FloatRange(1, 4, 0.5)):  # 调用__reversed__()方法，进行反迭代
    print x, " ",

四、对迭代器做切片操作

from itertools import islice
# islice切片对象，可以对可迭代对象进行切片
# islice(iterable, [start,] stop [, step]) --> islice object 
# 如果只有一个参数，表示从第一个开始第一个元素开始，如果想从某一个位置到最后，可以islice(iterable,40,None)方法

f=open('digital','r')

for x in islice(f,3,5,1):
    print x,

五、在for中迭代多个可迭代对象

from random import randint
from itertools import chain

# 要求统计三科总成绩
chinese = [randint(60, 100) for _ in xrange(40)]
math = [randint(60, 100) for _ in xrange(40)]
english = [randint(60, 100) for _ in xrange(40)]

total = []  # 总分列表
for c, m, e in zip(chinese, math, english):  # zip将三个可迭代对象合并成一个元组列表[(),()...]，如果三个对象长度不同，以最小的为准
    total.append(c + m + e)
print total

# 要求统三个班90分以上人数
c1 = [randint(60, 100) for _ in xrange(40)]
c2 = [randint(60, 100) for _ in xrange(45)]
c3 = [randint(60, 100) for _ in xrange(32)]
total = 0
for x in chain(c1, c2, c3):  # chain 将多个可迭代对象串联成一个可迭代对象
    if x > 90:
        total += 1