修改 __getitem__ 后无法对列表进行切片

Question

对于特定用例，我想定义一个列表类，如果索引为负数或超出范围，则返回 0。

我目前的方法已经服务于特定目的：

class mlist(list):
    def __getitem__(self, n):
        if (len(self)<=n) or (n<0):
            return 0
        return super(mlist, self).__getitem__(n)

l = mlist([1,2,3,4])
l[-2]
>>> 0
l[10]
>>> 0

但不幸的是，它在切片列表时会导致一些不良行为：

l[0:2]
>>> TypeError: '<=' not supported between instances of 'int' and 'slice'

有没有办法解决这个问题？

Answer 1

您想取消IndexError并返回0 。 所以，你也可以这样做：

class mlist(list):
    def __getitem__(self, n):
        try:
            return super().__getitem__(n)
        except IndexError:
            return 0

但是，这允许像通常列表一样进行负索引（从末尾访问）（因为这不会引发IndexError ）。 如果你也想抑制它，你可以尝试：

class mlist(list):
    def __getitem__(self, n):
        try:
            if n < 0 or n >= len(self):
                return 0
            return super().__getitem__(n)
        except TypeError:
            return super().__getitem__(n)

这假设n是一个 int 并相应地工作，如果不是，则会在比较时TypeError并调用super方法。 然后，这既处理切片，又在传递无效类型（例如字符串）时抛出适当的错误。

Answer 2

问题是参数n是您在方括号之间传递的任何内容，无论是索引（因此键入int ）还是切片（类型slice ）。

您当前的代码仅适用于n类型为int ，它不能正确处理切片。

如果您想保持切片功能与默认相同，为什么不调用super().__getitem__(n) ？

class mlist(list):
    def __getitem__(self, n):
        # If n is a slice
        if isinstance(n, slice):
            return super().__getitem__(n)

        # Otherwise
        if len(self)<=n or n<0:
            return 0
        return super(mlist, self).__getitem__(n)

但是，如果您想为切片实现自己的逻辑，则可以在if isinstance(n, slice):块中执行此操作。