《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
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