Zugriff auf einen Wörterbuchwert über benutzerdefinierten Objektwert in Python?

Question

Also habe ich ein Quadrat, das aus einer Reihe von Punkten besteht. Zu jedem Zeitpunkt gibt es einen entsprechenden Wert.

Was ich tun möchte, ist ein Wörterbuch wie dieses bauen:

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 

Allerdings, wenn ich später einen neuen Punkt-Objekt erstellen und versuchen, den Wert des Wörterbuchs für den Zugriff auf den Schlüssel dieses Punktes es doesn ‚t Arbeit ..

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

ich vermute, dass dies, weil python nicht zwei Objekte nicht berücksichtigt mit den gleichen Eigenschaften das gleiche Objekt sein? Gibt es einen Weg dazu? Danke

Answer 1

Definieren Sie Point.__hash__() und Point.__eq__(), damit sie innerhalb dicts richtig verglichen werden können.

Und während Sie dabei sind, überlegen Sie, Point.__repr__() zu definieren, so dass Sie anständig aussehende Darstellungen Ihrer Point Objekte erhalten.

Answer 2

Ja, definieren Sie die Methoden __eq__ und __hash__ für Ihre Point-Klasse.

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

Gibt es Gründe, nicht nur ein Tupel verwenden:

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

Answer 4

Schritte: eine benutzerdefinierte Schlüssel-Klasse implementieren und Hash und Gleichheit Funktion außer Kraft setzen.

z.B.

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) 

main-Methode

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] 

Siehe den Artikel Using custom class as key in python dictionary für nähere Details.