python - 在 __getitem__ 中实现切片

Question

我正在尝试为我正在创建的创建矢量表示的类实现切片功能。

到目前为止，我有这段代码，我相信它会正确实现切片，但是每当我执行像 v[4] 这样的调用时，其中 v 是一个向量，python 返回一个关于没有足够参数的错误.所以我想弄清楚如何在我的类中定义 getitem 特殊方法来处理普通索引和切片。

def __getitem__(self, start, stop, step):
    index = start
    if stop == None:
        end = start + 1
    else:
        end = stop
    if step == None:
        stride = 1
    else:
        stride = step
    return self.__data[index:end:stride]

Answer 1

__getitem__() 方法会在对象被切片时接收到一个 slice 对象。只需查看 slice 对象的 start、stop 和 step 成员，即可获取组件切片。

>>> class C(object):
...   def __getitem__(self, val):
...     print val
... 
>>> c = C()
>>> c[3]
3
>>> c[3:4]
slice(3, 4, None)
>>> c[3:4:-2]
slice(3, 4, -2)
>>> c[():1j:'a']
slice((), 1j, 'a')