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个Marys。如何才能使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__仅根据年龄定义。）

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

1 个答案: