python如何进行类的实现_python中class类的可迭代实现以及迭代函数

1. 当定义一个普通的类时,指向类的实例默认情况下是不可迭代的,如下

In [3]: from collections import Iterable

In [4]: class Fruit(object):

...: def __init__(self):

...: self.item = list()

...: def add(self,name):

...: self.item.append(name)

...:

In [5]: list1 = Fruit()

In [6]: isinstance(list1,Iterable)

Out[6]: False

2. 因此在这种情况下,当对一个类的实例想要使用for...in...循环进行遍历时,便会报不可迭代的错误,为此要对函数进行

改造,使其具有可迭代性,且能正确进行迭代操作,改造结果如下

class Fruit():

def __init__(self):

self.item = list()

# 为next函数计数器初始化

self.count = 0

def add(self,name):

self.item.append(name)

# 使函数具有可迭代性

def __iter__(self):

return self

def __next__(self):

# 获取下一个数

if self.count < len(self.item):

result = self.item[self.count]

self.count += 1

return result

else:

# 遍历结束后,抛出异常,终止遍历

raise StopIteration

list1 = Fruit()

list1.add("apple")

list1.add("banana")

list1.add("orange")

for item in list1:

print(item)

3. 下面详细解释,python在调用for...in...循环进行遍历时,对于类的实例的访问过程:

——检查该类是否具有迭代性,即调用类中的iter()方法

—— 通过iter()函数获取类中的迭代器,通过调用迭代器中的next()方法,获取下一个值,并赋值给result

——遇到StopIteration的异常后循环结束

4. 可将上段代码中对类的定义拆分成两个类来看,更加直观:

class Fruit():

def __init__(self):

self.item = list()

def add(self,name):

self.item.append(name)

# 使函数具有可迭代性

def __iter__(self):

return myIterator(self)

5. 以上是对Fruit类的定义,在其中定义iter()方法,使其具有可迭代性,通过iter()函数的返回值调用myIterator()迭代器;

并对该迭代器传递一个self参数,使迭代器可以指向到自身。

这部分代码使得类具有迭代性。

class myIterator():

def __init__(self,fru):

self.fru = fru

self.count = 0

def __next__(self):

if self.count < len(self.fru.item):

result = self.fru.item[self.count]

self.count += 1

return result

else:

raise StopIteration

6. 以上为Fruit类的迭代器通过参数fru的赋值,在迭代器中使self.fru变量指向Fruit类,从而在next函数中可以对Fruit类中

的实例进行访问控制 。

这部分代码完成了类的迭代器,保证可以调用next函数。