Python: Overloading the [] opeartor

Question

Hi,

I have been looking for the function name to override to overload the [] operator. Could anyone help?

Thanks

Answer 1

You need to use the __getitem__ method.

>>> class MyClass:
...     def __getitem__(self,index):
...         return index * 2
...
>>> myobj = MyClass()
>>> myobj[3]
6

And if you're going to be setting values you'll need to implement the __setitem__ method too, otherwise this will happen:

>>> myobj[5] = 1
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: MyClass instance has no attribute '__setitem__'

Answer 2

You are looking for the __getitem__ method. See http://docs.python.org/reference/datamodel.html, section 3.4.5

Answer 3

To fully overload it you also need to implement the __setitem__and __delitem__ methods.

edit

I almost forgot... if you want to completely emulate a list, you also need __getslice__, __setslice__ and __delslice__.

There are all documented in http://docs.python.org/reference/datamodel.html