我正在尝试创建一个继承自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你知道我怎样才能实现我想做的事吗?
发布于 2011-03-30 10:54:23
您可以简单地从__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 itemncoghlan建议使用生成器来代替生成器表达式,这使得调试更容易
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发布于 2011-03-30 09:30:55
__iter__返回一个iterator object。如果您需要在每次迭代中执行某些操作,则必须实现您自己的方法(它必须实现链接文档中描述的两个方法)。
发布于 2011-03-30 09:35:47
python类__iter__()方法实际上返回一个迭代器对象。请参阅以下内容作为参考:http://docs.python.org/library/stdtypes.html#iterator-types
在迭代器对象上,将在循环的每个步骤上调用next()方法。您可以编写一个自定义迭代器,它将由您的自定义列表返回。
https://stackoverflow.com/questions/5480898
复制相似问题