首页 > 代码库 > Python中实现iterator
Python中实现iterator
今天和大家分享的是python开发中实现iterator的方法步骤,希望可以帮助大家更好的学习和使用相关知识,一起来看看吧。
创建了一个实体类,大致如下:
class Account():
def __init__(self,
account_name,
account_type,
account_cost,
return_amount=0):
self.account_name = account_name # 账户名
self.account_type = account_type # 账户类型
self.account_cost = account_cost # 月结费用
self.return_amount = return_amount # 返还金额
然后创建一个实体列表:
accounts = [Account("张三", "年费用户", 450.00, 50),
Account("李四", "月结用户", 100.00),
Account("杨不悔", "月结用户", 190.00, 25),
Account("任我行", "月结用户", 70.00, 10),
Account("凌未风", "年费用户", 400.00, 40)]
我想要执行next()功能,也就是需要的时候“next”一下,得到List中的下一个元素。直接测试一下:
结果发现List不支持next()特性。这时候,List只是一个 iterable ,而不是 iterator 。 iterable 和 iterator 的区别如下:
. iterable —— 只实现了__iter__的对象;
. iterator —— 同时实现了__iter__和__next__方法的对象。
其中,__iter__返回 iterator 对象,__next__则返回迭代过程的下一个元素。
1. 让列表成为 iterator
要让前面的accounts List成为 iterator 只需简单的一个iter()函数:
accounts_iterator = iter(accounts)
(next(accounts_iterator)).account_name
结果如下图所示:
这么简单的函数,估计还是有不少Python开发者不知道吧?
2. 自定义 iterator 对象
扩展开来讲,如何定义自己的 iterator 对象呢?其实也就是按照上面的定义,实现__iter__和__next__方法。
我们接下来定义一个AccountIterator类:
class AccountIterator():
def __init__(self, accounts):
self.accounts = accounts # 账户集合
self.index = 0
def __iter__(self):
return self
def __next__(self):
if self.index >= len(self.accounts):
raise StopIteration("到头了...")
else:
self.index += 1
return self.accounts[self.index-1]
运行结果如:
通过这一阵折腾,next()功能就实现了。Python有不少意外的功能,还等着我们不断去探究,也许这就是Python的魅力及极客之处。
文章来源:瓜园耕读
Python中实现iterator