__getitem__在python中重载

Question

我试图理解python中的运算符重载，我编写了一个小程序，它重载__getitem__()方法并调用for循环：

class Ovl:
    data = [1,2,3,4,5]
    def __getitem__(self, index):
        return "Hiii"

x=Ovl()

for item in x:
    print item

该程序进入无限循环打印“Hiii”。 我想知道背后的原因。

Answer 1

根据object.__getitem__注意，

注意： for循环期望为非法索引引发IndexError ，以允许正确检测序列的结尾 。

>>> class Ovl:
...     data = [1,2,3,4,5]
...     def __getitem__(self, index):
...         if index >= 5:
...             raise IndexError('End')
...         return "Hiii"
...
>>> x=Ovl()
>>>
>>> for item in x:
...     print item
...
Hiii
Hiii
Hiii
Hiii
Hiii

Answer 2

__getitem__的重载实际上允许你做这样的事情：

>>> x = Ovl()
>>> x[6]
'Hi'
>>> x[8]
'Hi'

如果你想用for实现迭代你的类，你需要重载__iter__方法（参见这里： 如何为容器对象（Python）实现__iter __（self） ）。

循环运行无限的原因是__iter__的默认实现显然在内部调用了__getitem__ 。 实际上，你的for循环调用__getitem__(0) ， __getitem__(1) ， __getitem__(2) ， ...顺序，所以它似乎运行无限。

如果要修改__getitem__方法：

def __getitem__(self, index):
    return index, "Hi"

你可以看到到底发生了什么：

>>> x = Ovl()
>>> for item in x:
    print item 

(0, 'Hi')
(1, 'Hi')
(2, 'Hi')
(3, 'Hi')
(4, 'Hi')
(5, 'Hi')
(6, 'Hi')
...