python - 对一个类进行多重索引？

Question

这个问题已经有答案了:

Possible to use more than one argument on __getitem__? (5 个回答)

已关闭 4 年前。

我有以下代码:

class C:
    def __getitem__(self,a,b):
        return 1

c = C()
c[2,3]

TypeError: Missing 1 required positional argument : b

我的问题是在索引 c 时如何提供第二个参数？

Answer 1

没有办法直接在__getitem__中处理这个问题 (docs) ;您必须让 c[x] 返回某种子对象，您可以使用第二个 inxed [y] 来获取单个项目。您可能想要的是让 __getitem__ 接受具有此类坐标的元组:

def __getitem__(self, tuple):
    y, x = tuple
    return self.matrix[y][x]

然后执行:

c[x, y]