我正在尝试创建一个继承自python列表的类。我希望列表的每个循环初始化/最终确定列表的元素。我认为这可以通过覆盖python列表的__iter__
方法来完成,但我似乎无法让它工作。 __iter__
方法似乎只调用一次? (见下文)
class MyList(list):
def __iter__(self):
print 'do something'
return list.__iter__(self)
my_list = MyList(range(10))
print my_list
for item in my_list:
print item
输出
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
do something
0
1
2
3
4
5
6
7
8
9
知道如何实现我想做的事情吗?
答案 0 :(得分:10)
您只需从__iter__()
class MyList(list):
def __iter__(self):
return (self.do_something(x) for x in list.__iter__(self))
def do_something(self, x):
print 'do something', x
return x
my_list = MyList(range(10))
print my_list
for item in my_list:
print item
ncoghlan建议使用生成器代替生成器表达式,以便于调试
class MyList(list):
def __iter__(self):
for x in list.__iter__(self):
yield self.do_something(x)
def do_something(self, x):
print 'do something', x
return x
my_list = MyList(range(10))
print my_list
for item in my_list:
print item
或者你可以在这里使用imap
from itertools import imap
class MyList(list):
def __iter__(self):
return imap(self.do_something, list.__iter__(self))
def do_something(self, x):
print 'do something', x
return x
my_list = MyList(range(10))
print my_list
for item in my_list:
print item
答案 1 :(得分:3)
__iter__
返回iterator object。如果您需要在每次迭代中执行某些操作,则必须实现自己的(它必须实现链接文档中描述的两种方法)。
答案 2 :(得分:2)
python类__iter__()
方法实际上返回一个迭代器对象。
请参阅以下内容以供参考:http://docs.python.org/library/stdtypes.html#iterator-types
在迭代器对象上,将在循环的每一步调用next()
方法。您可以编写一个自定义迭代器,它将由您的自定义列表返回。
答案 3 :(得分:0)
我想用列表的每个循环
初始化/最终确定列表的元素
列表无法直接控制此情况。几个循环可以并行访问列表(例如,在嵌套循环中)。每个循环都由迭代器控制,每次调用时__iter__()
都应该返回迭代器。你不能一直返回相同的迭代器实例,因为所说的并行循环会相互干扰。
但是在迭代器中,您可以根据需要自由地实例化项目。从迭代器内部完成它们是不明智的,因为一旦你返回一个项目,你就失去了对它的控制。如果您稍后完成它,仍然依赖于该项目的外部代码可能会失败。如果您需要最终确定项目,请将其作为项目消费者的责任。
答案 4 :(得分:-2)
我不明白你到底想要做什么,但你已经被告知的是真的。 next()
是每次迭代中调用的方法。
class MyList(list):
def next(self):
print "Do something"
return list.next(self)
(我没有测试过它)