¿Acceder a un valor de diccionario por valor de objeto personalizado en Python?

Question

Así que tengo un cuadrado que se compone de una serie de puntos. En cada punto hay un valor correspondiente.

Lo que quiero hacer es construir un diccionario de la siguiente manera:

class Point: 
    def __init__(self, x, y): 
     self._x = x 
     self._y = y 


square = {}  
for x in range(0, 5): 
     for y in range(0, 5): 
      point = Point(x,y) 
      square[point] = None 

Sin embargo, si más adelante se crea un nuevo objeto de punto y tratar de acceder al valor del diccionario con la clave de ese punto que doesn 't trabajo ..

>> square[Point(2,2)] 
Traceback (most recent call last): 
    File "<pyshell#19>", line 1, in <module> 
    square[Point(2,2)] 
KeyError: <__main__.Point instance at 0x02E6C378> 

supongo que esto se debe pitón no tiene en cuenta dos objetos con las mismas propiedades que ser el mismo objeto? ¿Hay alguna forma de evitar esto? Gracias

Answer 1

Defina Point.__hash__() y Point.__eq__() para que se puedan comparar correctamente dentro de los dicts.

Y mientras lo hace, considere la definición de Point.__repr__() para que obtenga representaciones de aspecto decente de sus objetos Point.

Answer 2

Sí, defina los métodos __eq__ y __hash__ en su clase Point.

class Point: 
    def __init__(self, x, y): 
     self._x = x 
     self._y = y 

    def __eq__(self, other): 
     return self._x == other._x and self._y == other._y 

    def __hash__(self): 
     #This one's up to you, but it should be unique. Something like x*1000000 + y. 

Answer 3

algún motivo no sólo tiene que utilizar una tupla:

>>> s={} 
>>> s[1,2]=5 
>>> s 
{(1, 2): 5} 
>>> s[1,2] 
5 

Answer 4

pasos: Implementar una clase clave personalizada y anulan la función hash y la igualdad.

p. Ej.

class CustomDictKey(object): 

def __init__(self, 
      param1, 
      param2): 

      self._param1 = param1 
      self._param2 = param2 

# overriding hash and equality function does the trick 

def __hash__(self): 
    return hash((self._param1, 
      self._param2)) 

def __eq__(self, other): 
    return ((self._param1, 
      self._param2) == (other._param1, 
      other._param2)) 

def __str__(self): 
    return "param 1: {0} param 2: {1} ".format(self._param1, self._param2) 

principal método

if name == 'main': 

    # create custom key 
    k1 = CustomDictKey(10,5) 

    k2 = CustomDictKey (2, 4) 

    dictionary = {} 

    #insert elements in dictionary with custom key 
    dictionary[k1] = 10 
    dictionary[k2] = 20 

    # access dictionary values with custom keys and print values 
    print "key: ", k1, "val :", dictionary[k1] 
    print "key: ", k2, "val :", dictionary[k2] 

remitir el artículo Using custom class as key in python dictionary para más detalles.