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 对象切成薄片时。只需看看 start, stop, ， 和 step 成员 slice 对象为了获取切片的组件。

>>> 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')

Answer 2

我有一个“综合”列表（数据比您想在内存中创建的数据大）和我的 __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."

切片不返回相同的类型，这是不可以的，但对我有用。

Answer 3

如何定义getItem类以处理普通索引和切片？

当您在下标表示法中使用结肠时，将自动创建切片对象 - 和 那 是传递给的 __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

Python 2，请注意：

在Python 2中，在分类一些内置类型时，您可能需要覆盖一种弃用的方法。

来自 Datamodel文档:

object.__getslice__(self, i, j)

自版本2.0以来被弃用： 支持切片对象作为参数 __getitem__() 方法。 （但是，CPYTHON中的内置类型目前仍在实施 __getslice__(). 。因此，在实施切片时，您必须在派生的类中覆盖它。）

这在Python 3中消失了。

Answer 4

正确的方法是拥有 __getitem__ 采用一个参数，可以是一个数字，也可以是切片对象。

看：

http://docs.python.org/library/functions.html#slice

http://docs.python.org/reference/datamodel.html#object.__getitem__

Answer 5

为了扩展亚伦的回答，例如 numpy, ，您可以通过检查是否进行多维切片 given 是一个 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)))