python - 如何创建一个也是列表的类？

Question

我想在 Python 中创建一个具有各种属性和方法的类，但要让它继承列表的功能，这样我就可以将对象附加到对象本身，而不是它的任何属性。我希望能够说“graph[3]”，而不是“graph.node_list[3]”。有办法做到这一点吗？

Answer 1

您真正需要做的就是提供一个 __getitem__

In [1]: class Foo:
   ...:     def __init__(self, *args):
   ...:         self.args = args
   ...:     def __getitem__(self, i):
   ...:         return self.args[i]
   ...:     

In [2]: c = Foo(3,4,5)

In [3]: c[2]
Out[3]: 5

In [4]: c[3]
IndexError: tuple index out of range #traceback removed for brevity

In [5]: for i in c: print(i) #look, ma, I can even use a for-loop!
3
4
5

附录:您可能还想提供其他方法。 __len__ 绝对是其中之一。魔术方法的列表相当长，我建议仔细阅读它们并选择有意义的方法。