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

小编典典

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

python

我有一个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个Mary。如何使Person具有相同年龄的实例被视为同一个人，并且仅添加到集合一次？

换句话说，我如何得到的结果{<Person tom>, <Person mary>}？

阅读 225

2020-12-20

共1个答案

小编典典

当将新对象添加到python集中时，首先计算该对象的哈希码，然后，如果该集中已经存在一个或多个具有相同哈希码的对象，则将这些对象与新对象。

这样的结果是您需要在类上实现__hash__(...)and__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__仅根据年龄来定义）。

2020-12-20