每个循环调用python list __iter__方法?

时间:2011-03-30 01:26:08

标签: python iterator loops python-2.6

我正在尝试创建一个继承自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

知道如何实现我想做的事情吗?

5 个答案:

答案 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)

(我没有测试过它)