哪些运算符要重载以使 Python 设置为正常运行 [重复]

Question

我正在编写一个简单的容器类，我想将其实例存储在set 中，并希望删除重复项。比如我可以用tuple作为容器来写：

in>  set([(1,2),(1,2)])
out> {(1,2)}

但如果我改为定义

class Point(object):

    def __init__(self, x, y):
        self.x = x
        self.y = y

    def __ge__(self, other):
        if self.x > other.x:
            return True
        elif self.x == other.x:
            return self.y >= other.y
        else:
            return False

    def __le__(self, other):
        if self.x < other.x:
            return True
        elif self.x == other.x:
            return self.y <= other.y
        else:
            return False

    def __eq__(self, other):
        return self.x == other.x and self.y == other.y

然后尝试

set([Point(1,2), Point(1,2)])

我最终得到了一组 2 个对象而不是 1 个。我需要重载哪些运算符（或者还需要做什么）才能使 set 以可预测的方式运行？

谢谢。

Answer 1

根据https://docs.python.org/2.7/library/stdtypes.html#set-types-set-frozenset，

集合对象是不同的可散列对象的无序集合。

根据https://docs.python.org/2.7/glossary.html#term-hashable，

如果一个对象的哈希值在其生命周期内永远不会改变（它需要__hash__() 方法），并且可以与其他对象进行比较（它需要__eq__() 或__cmp__() 方法），则它是可哈希的。

你有__eq__，所以你现在只需要__hash__。 （__ne__ 也应该实现，否则你会得到x == y 与not (x != y) 不匹配的结果。）

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

    def __eq__(self, other):
        if isinstance(other, Point):
            return self.x == other.x and self.y == other.y
        return NotImplemented

    def __ne__(self, other)
        if isinstance(other, Point):
            return not (self == other)
        return NotImplemented

    def __hash__(self):
        return hash((self.x, self.y))


print(set([Point(1,2), Point(1,2)]))

结果：

set([<__main__.Point object at 0x02F4F090>])

对于样式点，您可能还想实现__repr__，这样您的设置对象看起来不错。添加def __repr__(self): return "Point({}, {})".format(self.x, self.y)，您的集合将显示为set([Point(1, 2)])。