python - 为用户定义的类设置多重性

Question

我想在我制作的一些类(class)中使用集合。我希望这些集合能够限制该类对象的多样性。但是我有一个问题。考虑这个玩具示例:

class Thing(object):

    def __init__(self, value):
        self.value = value

    def __eq__(self, other):
        return self.value == other.value

    def __repr__(self):
        return str(self.value)

a = Thing(5)
b = Thing(5)

s = set()
s.add(a)
s.add(b)
print s

这段代码的输出是:

set([5, 5])

我只是假设 set() 会使用 __eq__ 来确定是否应该包含该对象，但它可能使用 is。是否有一个简单的解决方法，使输出只是 set([5])？

Answer 1

您需要定义类的__hash__ 方法以返回基于value 的哈希码。换句话说，你需要让你的类(class) hashable .

class Thing(object):

    def __init__(self, value):
        self.value = value

    def __eq__(self, other):
        return self.value == other.value

    def __repr__(self):
        return str(self.value)

    def __hash__(self):
        return hash(self.value)

a = Thing(5)
b = Thing(5)

s = set()
s.add(a)
s.add(b)
print s