Python项目实战之迭代器实现字符串的逆序输出

《Python迭代器》一节已经对如何创建迭代器做了详细的介绍,本节将利用迭代器完成对字符串的逆序操作。项目要求是这样的,定义一个类,要求在实现迭代器功能的基础上,能够对用户输入的字符串做逆序输出操作。

实现思路是这样的,自定义一个类并重载其 __init__() 初始化方法,实现为自身私有成员赋值。同时重载 __iter__() 和 __next__() 方法,使其具有迭代器功能。在此基础上,如果想实现对用户输入的字符串进行逆序输出,就需要在 __next__() 方法中实现从后往前返回字符。

实现代码如下:

class Reverse:
    def __init__(self, string):
        self.__string = string
        self.__index = len(string)
    def __iter__(self):
        return self
    def __next__(self):
        self.__index -= 1
        return self.__string[self.__index]
revstr = Reverse('Python')
for c in revstr:
    print(c,end=" ")

运行结果为:

n o h t y P n o h t y P Traceback (most recent call last):
  File "C:\Users\mengma\Desktop\demo.py", line 11, in <module>
    for c in revstr:
  File "C:\Users\mengma\Desktop\demo.py", line 9, in __next__
    return self.__string[self.__index]
IndexError: string index out of range

可以看到,上面程序在逆序输出两遍"python"的同时,Python解释器报出 IndexError 错误,这是什么原因呢?

很简单,因为程序没有设置遍历的终止条件,换句话说,没有对 __index 私有变量的值对限制,这里 __index 的取值范围应为(-len(self.__index), len(self.__index)),这也是导致上面程序运行结果的根本原因。

编写迭代器最容易忽视的一个环节,就是在自定义类中加入对循环结束的判断,并抛出 StopIteration 异常,只有这么做了,for 循环才会接收到 StopIteration 异常,并当做终止信号来结束循环。

所以,我们需要对上面程序做适当的调整,如下所示:

class Reverse:
    def __init__(self, string):
        self.__string = string
        self.__index = len(string)
    def __iter__(self):
        return self
    def __next__(self):
        if self.__index == 0:
            raise(StopIteration)
        self.__index -= 1
        return self.__string[self.__index]
revstr = Reverse('Python')
for c in revstr:
    print(c,end=" ")

运行结果为:

n o h t y P