Python中实现iterator

Posted 2020-08-17

今天和大家分享的是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的魅力及极客之处。

 

文章来源：瓜园耕读