将对象添加到python的集合中并通过对象的属性确定

Question

我有一个像这样的Person 类：

class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age

    def __repr__(self):
        return '<Person {}>'.format(self.name)

我想将此类的一些实例添加到集合中，如下所示：

tom = Person('tom', 18)
mary = Person('mary', 22)
mary2 = Person('mary2', 22)

person_set = {tom, mary, mary2}
print(person_set)
# output: {<Person tom>, <Person mary>, <Person mary2>}

如您所见，集合中有 2 个玛丽。我怎样才能使具有相同年龄的Person 实例被视为同一个人，并且只添加到集合中一次？

换句话说，我怎样才能得到{<Person tom>, <Person mary>}的结果？

Answer 1

当一个新对象被添加到 Python 集合中时，首先计算对象的哈希码，然后，如果集合中已经有一个或多个具有相同哈希码的对象，则测试这些对象与新对象相等。

这样做的结果是你需要在你的类上实现__hash__(...) 和__eq__(...) 方法。例如：

class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age

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

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

    def __repr__(self):
        return '<Person {}>'.format(self.name)

tom = Person('tom', 18)
mary = Person('mary', 22)
mary2 = Person('mary2', 22)

person_set = {tom, mary, mary2}
print(person_set)
# output: {<Person tom>, <Person mary>}

但是，您应该非常仔细地考虑 __hash__ 和 __eq__ 的正确实现对于您的班级应该是什么。上面的示例有效，但没有意义（例如，__hash__ 和 __eq__ 均仅根据年龄定义）。