Đối tượng của loại tùy chỉnh làm khóa từ điển

Tôi phải làm gì để sử dụng các đối tượng thuộc loại tùy chỉnh làm khóa trong từ điển Python (nơi tôi không muốn "id đối tượng" đóng vai trò là khóa), vd

class MyThing:
    def __init__(self,name,location,length):
            self.name = name
            self.location = location
            self.length = length

Tôi muốn sử dụng MyThing như các khóa được coi là giống nhau nếu tên và vị trí giống nhau. Từ C # / Java tôi đã quen với việc ghi đè và cung cấp phương thức bằng và mã băm, và hứa sẽ không làm thay đổi bất cứ điều gì mà mã băm phụ thuộc vào.

Tôi phải làm gì trong Python để thực hiện điều này? Tôi có nên thậm chí?

(Trong một trường hợp đơn giản, như ở đây, có lẽ tốt hơn là chỉ đặt một (tên, vị trí) tuple làm khóa - nhưng hãy xem xét tôi muốn khóa là một đối tượng)

Bạn cần thêm 2 phương thức , lưu ý __hash__và __eq__:

class MyThing:
    def __init__(self,name,location,length):
        self.name = name
        self.location = location
        self.length = length

    def __hash__(self):
        return hash((self.name, self.location))

    def __eq__(self, other):
        return (self.name, self.location) == (other.name, other.location)

    def __ne__(self, other):
        # Not strictly necessary, but to avoid having both x==y and x!=y
        # True at the same time
        return not(self == other)

Tài liệu dict Python xác định các yêu cầu này trên các đối tượng chính, tức là chúng phải được băm .

Một cách khác trong Python 2.6 trở lên là sử dụng collections.namedtuple()- nó giúp bạn tiết kiệm bằng cách viết bất kỳ phương thức đặc biệt nào:

from collections import namedtuple
MyThingBase = namedtuple("MyThingBase", ["name", "location"])
class MyThing(MyThingBase):
    def __new__(cls, name, location, length):
        obj = MyThingBase.__new__(cls, name, location)
        obj.length = length
        return obj

a = MyThing("a", "here", 10)
b = MyThing("a", "here", 20)
c = MyThing("c", "there", 10)
a == b
# True
hash(a) == hash(b)
# True
a == c
# False

Bạn ghi đè __hash__nếu bạn muốn ngữ nghĩa băm đặc biệt và __cmp__hoặc __eq__để làm cho lớp của bạn có thể sử dụng làm khóa. Các đối tượng so sánh bằng nhau cần có cùng giá trị băm.

Python hy vọng sẽ __hash__trả về một số nguyên, Banana()không nên trả về :)

Các lớp __hash__do người dùng định nghĩa có mặc định gọi id(self)như bạn đã lưu ý.

Có một số lời khuyên bổ sung từ tài liệu .:

Các lớp kế thừa một __hash__() phương thức từ một lớp cha nhưng thay đổi ý nghĩa của __cmp__()hoặc __eq__() giá trị băm được trả về không còn phù hợp nữa (ví dụ: bằng cách chuyển sang một khái niệm đẳng thức dựa trên giá trị thay vì đẳng thức dựa trên danh tính mặc định) có thể tự gắn cờ rõ ràng là không thể phá vỡ bằng cách thiết lập __hash__ = None định nghĩa lớp. Làm như vậy có nghĩa là không chỉ các thể hiện của lớp sẽ tăng TypeError thích hợp khi một chương trình cố gắng truy xuất giá trị băm của chúng, mà chúng cũng sẽ được xác định chính xác là không thể xóa được khi kiểm tra isinstance(obj, collections.Hashable) (không giống như các lớp xác định __hash__()rõ ràng để tăng TypeError).