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


112

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

Я маю цей код до сих пір, і я вважаю, що він належним чином реалізує фрагмент, але щоразу, коли я виконую дзвінок, наприклад, 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]

Відповіді:


118

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

10
Примітка: для розширення типів вбудованих файлів, таких як список або кортеж, ви повинні реалізувати __getslice__для версій python 2.X. дивись docs.python.org/2/reference/datamodel.html#object.__getslice__
gregorySalvan

@gregorySalvan: Чи не повторюється той приклад сумісності під цим розділом?
Ерік

3
@Eric: Ні, тому що присутність другої кишки обходить __get/set/delslice__. Хоча це дуже тонко.
user2357112 підтримує Моніку

@ user2357112: Нічого, повністю пропустив цю другу кишку - спасибі!
Ерік

@alancalvitti IIRC, це для створення класів нового стилю в Python 2.
wjandrea

64

У мене є "синтетичний" список (той, де даних більше, ніж ви хотіли б створити в пам'яті), і моє __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."

Фрагмент не повертає того ж типу, який є ні-ні, але він працює для мене.


1
Чи не слід, якщо ключ> = len (само) бути, якщо ключ <0 або ключ> = len (само)? Що робити, якщо ключ <-len (self) передано?
Естан

20

Як визначити клас getitem для обробки як простих індексів, так і зрізів?

Шматочок об'єкти отримує автоматично створюється , коли ви використовуєте двокрапка індексного позначення - і що це те , що передається __getitem__. Використовуйте isinstanceдля перевірки наявності у вас об'єкта фрагмента:

from __future__ import print_function

class Sliceable(object):
    def __getitem__(self, subscript):
        if isinstance(subscript, slice):
            # do your handling for a slice object:
            print(subscript.start, subscript.stop, subscript.step)
        else:
            # Do your handling for a plain index
            print(subscript)

Скажімо, ми використовували об’єкт діапазону, але ми хочемо, щоб фрагменти повертали списки замість нових об'єктів діапазону (як це відбувається):

>>> range(1,100, 4)[::-1]
range(97, -3, -4)

Ми не можемо діапазон підкласів через внутрішні обмеження, але ми можемо делегувати його:

class Range:
    """like builtin range, but when sliced gives a list"""
    __slots__ = "_range"
    def __init__(self, *args):
        self._range = range(*args) # takes no keyword arguments.
    def __getattr__(self, name):
        return getattr(self._range, name)
    def __getitem__(self, subscript):
        result = self._range.__getitem__(subscript)
        if isinstance(subscript, slice):
            return list(result)
        else:
            return result

r = Range(100)

У нас немає ідеально змінного об'єкта Range, але він досить близький:

>>> r[1:3]
[1, 2]
>>> r[1]
1
>>> 2 in r
True
>>> r.count(3)
1

Щоб краще зрозуміти позначення фрагмента, ось приклад використання Sliceable:

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


7

Щоб розширити відповідь Аарона, на такі речі, як 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)))

Як незначне подальше спостереження, ось приклад використання цього для зіставлення між індексуванням MATLAB та індексуванням NumPy (яке в даний час не підтримується в MATLAB R2016b), з прикладом його використання .
Ерік Кузен

Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.