列表/数组范围之外的零

Question

使用 Python 列表

L=[1,2,3,4]

如果m 与0,1,2,3 不同，我希望L[m] = 0，即：

...
L[-2]=0 
L[-1]=0  
L[0]=1
L[1]=2
L[2]=3
L[3]=4
L[4]=0
L[5]=0

和

L[-2:2] = [0, 0, 1, 2]

这不适用于经典列表或数组。这样做的好方法是什么？

编辑：这是一个不错的解决方案（由此处给出的答案给出）：

class MyList(list):
    def __getitem__(self, index):
        return super(MyList, self).__getitem__(index) if index >= 0 and index < len(self) else 0

但我仍然无法拥有

L[-2:2] = [0, 0, 1, 2]

Answer 1

解决它的一个问题是声明你自己的get函数

def get(l, p):
   try:
       return l[p]
   except IndexError:
       return 0

ofc 最后一行可以是 l.append(0) 或其他的东西

Answer 2

列表 len 中的随机索引仅适用于切片。如果要获取（或设置）某些对象，则必须仅在列表范围内使用索引。 所以

L[4]=0
x=L[4]

没用，但是

L[4:6]=0,0 or L[4:6]=[0,0]
x,y=L[2:10000] # x=3,y=4

会起作用

Answer 3

您可以将L 转换为dict：

In [1]: L=[1,2,3,4]

In [2]: D=dict([(x, y) for x, y in enumerate(L)])

In [3]: [D.get(i, 0) for i in xrange(-3, 5)]
Out[3]: [0, 0, 0, 1, 2, 3, 4, 0]

Answer 4

你可以使用一个特殊的get函数：

def get(l, p):
    return l[p] if p >= 0 and p < len(l) else 0

甚至重写__getitem__ 方法：

class MyList(list):
    def __getitem__(self, index):
        return super(MyList, self).__getitem__(index) if index >= 0 and index < len(self) else 0

例子：

>>> l2 = MyList([3, 5, 6, 8])
>>> l2[-1]
0
>>> l2[5]
0
>>> l2[2]
6

Answer 5

您可以在访问列表项时对内置列表进行子类化以提供默认值：

class MyList(list):
    def __getitem__(self, item):
        if isinstance(item, slice):
            step = item.step if item.step else 1
            return [self.__getitem__(i) for i in xrange(item.start, item.stop, step)]
        try:
            value = super(MyList, self).__getitem__(item)
        except IndexError:
            value = 0
        return value

    def __getslice__(self, start, stop):
        return self.__getitem__(slice(start, stop, None))

示例用法：

>> L = MyList([1,2,3,4])
>> L[0]
1
>> L[1]
2
>> L[2]
3
>> L[3]
4
>> L[4]
0
>> L[5]
0
>> L[0:6]
[1, 2, 3, 4, 0, 0]

归功于How to override the slice functionality of list in its derived class

Answer 6

您可以使用slice 对象，当使用“扩展切片”时，这些对象不是传递给__getslice__，而是传递给__getitem__。然后，将切片的start 移动到0，将stop 移动到len - 1，同时保持跟踪。然后加零：

class MyList(list):
    def __getitem__(self, item):
        if isinstance(item, slice):
            s, e = item.start, item.stop
            l = len(self) - 1
            left = -s if s < 0 else 0
            s = max(s, 0)
            right = e - l if e > l else 0
            e = min(e, l)
            return [0]*left + super(MyList, self).__getitem__(slice(s,e,item.step)) + [0]*right
        elif item < 0 or item >= len(self):
            return 0
        else:
            return super(MyList, self).__getitem__(item)

问题是：您必须强制您的 getslice 调用发送一个 slice 对象，您可以通过以下两种方式之一进行。

>>> a[-2:2:1]   # the step = 1 is necessary
[0, 0, 1, 2]

或

>>> a[slice(-2,2)]
[0, 0, 1, 2]

在两端都有效：

>>> a[-2:6:1]
[0, 0, 1, 2, 3, 0, 0, 0]

---

最初的尝试

如果__getslice__ 传递了a[-2:2] 给出的实际参数，那么这将起作用：

class MyList(list):
    def __getitem__(self, item):
        if item < 0 or item >= len(self):
            return 0
        return super(MyList, self).__getitem__(item)

    def __getslice__(self, s, e):
        print "input: ", s, e
        l = len(self) - 1
        left = -s if s < 0 else 0
        s = max(s, 0)
        right = e - l if e > l else 0
        e = min(e, l)

        return [0]*left + super(MyList, self).__getslice__(s,e) + [0]*right

但由于某种原因，a[-2:2] 调用 a.__getslice(2,2) 时两个值都是正值。

>>> a[-2:2]
input: 2 2