Python：在__getitem__中实现切片
我正在尝试为我创build的类创buildvector表示的切片function。
我有这个代码到目前为止，我相信会正确地实现切片，但每当我做一个像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]
__getitem__()方法将在slice对象时接收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')
我有一个“合成”列表（其中数据比你想要在内存中创build的更大），我的__getitem__如下所示：
def __getitem__( self, key ) : if isinstance( key, slice ) : #Get the start, stop, and step from the slice return [self[ii] for ii in xrange(*key.indices(len(self)))] elif isinstance( key, int ) : if key < 0 : #Handle negative indices key += len( self ) if key < 0 or key >= len( self ) : raise IndexError, "The index (%d) is out of range."%key return self.getData(key) #Get the data from elsewhere else: raise TypeError, "Invalid argument type."
切片不返回相同的types，这是一个不，不，但它适用于我。
正确的做法是让__getitem__取一个参数，它可以是一个数字，也可以是一个切片对象。
看到：
http://docs.python.org/library/functions.html#slice
http://docs.python.org/reference/datamodel.html#object.__ getitem __
如何定义getitem类来处理普通索引和切片？
在下标符号中使用冒号时，会自动创build切片对象 – 这就是传递给__getitem__ 。 使用isinstance来检查是否有切片对象：
from __future__ import print_function class Sliceable(object): def __getitem__(self, given): if isinstance(given, slice): # do your handling for a slice object: print(given.start, given.stop, given.step) else: # Do your handling for a plain index print(given)
用法示例：
>>> sliceme = Sliceable() >>> sliceme[1] 1 >>> sliceme[2] 2 >>> sliceme[:] None None None >>> sliceme[1:] 1 None None >>> sliceme[1:2] 1 2 None >>> sliceme[1:2:3] 1 2 3 >>> sliceme[:2:3] None 2 3 >>> sliceme[::3] None None 3 >>> sliceme[::] None None None >>> sliceme[:] None None None
为了扩展Aaron的答案，对于像numpy这样的东西，可以通过检查给given的tuple来检查tuple ：
class Sliceable(object): def __getitem__(self, given): if isinstance(given, slice): # do your handling for a slice object: print("slice", given.start, given.stop, given.step) elif isinstance(given, tuple): print("multidim", given) else: # Do your handling for a plain index print("plain", given) sliceme = Sliceable() sliceme[1] sliceme[::] sliceme[1:, ::2]
“`
输出：
('plain', 1) ('slice', None, None, None) ('multidim', (slice(1, None, None), slice(None, None, 2)))
