Python: Implementando slicing en __getitem__

Question

Estoy tratando de implementar la funcionalidad de corte para una clase que estoy creando que crea una representación vectorial.

Tengo este código hasta el momento, que creo que implementará correctamente la división pero cada vez que hago una llamada como v[4] donde v es un vector python devuelve un error acerca de no tener suficientes parámetros. Así que estoy tratando de encontrar la manera de definir el método especial getitem en mi clase para manejar tanto los índices simples como el corte.

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

El método __getitem__() recibirá un objeto slice cuando se corta el objeto. Simplemente mire los miembros start, stop y step del objeto slice para obtener los componentes del sector.

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

La forma correcta de hacer esto es tener __getitem__ tomar un parámetro, que puede ser un número, o un objeto de división.

Ver:

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

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

Answer 3

Tengo una lista de "sintético" (uno donde los datos es mayor de lo que se desee crear en la memoria) y mi __getitem__ parece esto:

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

La rebanada no devuelve el mismo tipo, que es un no- no, pero funciona para mí

Answer 4

¿Cómo se define la clase getitem para manejar los índices simples y el corte?

objetos de división se crea automáticamente cuando se utiliza una coma en la notación de subíndice - y que es lo que se pasa a __getitem__. Utilice isinstance para comprobar si tiene un objeto de división: el uso

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) 

Ejemplo:

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

Answer 5

Para extender la respuesta de Aarón, para cosas como numpy, que puede hacer el corte en lonchas multidimensional mediante la comprobación para ver si given es una 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] 

`` `

Salida:

('plain', 1) 
('slice', None, None, None) 
('multidim', (slice(1, None, None), slice(None, None, 2)))