- 当定义一个普通的类时,指向类的实例默认情况下是不可迭代的,如下
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
- 因此在这种情况下,当对一个类的实例想要使用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)
- 下面详细解释,python在调用for…in…循环进行遍历时,对于类的实例的访问过程:
——检查该类是否具有迭代性,即调用类中的iter()方法
—— 通过iter()函数获取类中的迭代器,通过调用迭代器中的next()方法,获取下一个值,并赋值给result
——遇到StopIteration的异常后循环结束
- 可将上段代码中对类的定义拆分成两个类来看,更加直观:
class Fruit():
def __init__(self):
self.item = list()
def add(self,name):
self.item.append(name)
# 使函数具有可迭代性
def __iter__(self):
return myIterator(self)
- 以上是对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
- 以上为Fruit类的迭代器通过参数fru的赋值,在迭代器中使self.fru变量指向Fruit类,从而在next函数中可以对Fruit类中的实例进行访问控制 。这部分代码完成了类的迭代器,保证可以调用next函数。