python迭代-如何实现反向迭代

如何实现反向迭代

问题举例

实现一个连续浮点数发生器FloatRange，根据给定范围（start， end）和步进值（step）

产生一系列连续的浮点数，如FloatRange（3.0， 4.0， 0.2）可产生序列：

正向：3.0>3.2>3.4>3.6>3.8>4.0

反向：4.0>3.8>3.6>3.4>3.2>3.0

 

分析

（1）列表的reverse()方法

list1 = [1, 2, 3, 4, 5]
list1.reverse()
for x in list1:
    print(x) # 5 4 3 2 1

改变了原列表，在某些情况下是不被允许的

 

（2）切片操作

list1 = [1, 2, 3, 4, 5]
list2 = list1[::-1]
for x in list2:
    print(x) # 5 4 3 2 1

创建和原列表等大的列表，浪费内存空间

 

解决思路

实现反向迭代协议的__reversesd__方法，它返回一个反向迭代器

 

代码

from decimal import Decimal

class FloatRange:
    def __init__(self, a, b, step):
        self.a = Decimal(str(a))
        self.b = Decimal(str(b))
        self.step = Decimal(str(step))

    def __iter__(self):
        t = self.a
        while t <= self.b:
            yield float(t)
            t += self.step

    def __reversed__(self):
        t = self.b
        while t >= self.a:
            yield float(t)
            t -= self.step

fr = FloatRange(3.0, 4.0, 0.2)
for x in fr:
    print(x)
print('-' * 20)
for x in reversed(fr):
    print(x)

 

参考资料：python3实用编程技巧进阶