Python: Реализация нарезки в __getitem__

Question

Я пытаюсь реализовать функциональность среза для создаваемого мной класса, который создает векторное представление.

Пока у меня есть этот код, который, я считаю, будет правильно реализовывать фрагмент, но всякий раз, когда я делаю вызов, такой как v[4], где v - векторный питон, выдает ошибку о нехватке параметров. Поэтому я пытаюсь выяснить, как определить специальный метод 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 есть устаревший метод, который вам может потребоваться переопределить при создании подклассов некоторых встроенных типов.

Из документации модели данных :

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