Python中set和dict底层实现原理——重写相等性判断

创建一个类，然后创建三个对象。

class Person:

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

    def __repr__(self):
        return f'{self.name}: {self.age}'


p1 = Person('勤勤', 20)
p2 = Person('勤勤', 20)
p3 = Person('勤勤', 18)
persons = {p1, p2, p3}
print(len(persons))
print(persons)

# 3
# {勤勤: 20, 勤勤: 18, 勤勤: 20}

即使对象的名字，年龄都相同。但是集合依旧认为它们是三个不同的元素，因为它们底层的哈希码是不同的。想要使名字相同并且年龄相同的对象使集合认为是相同的元素，需要重写__hash__和__eq__方法。

class Person:

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

    def __repr__(self):
        return f'{self.name}: {self.age}'
        
	# 哈希码不相同的对象，集合会认定它们是不同的元素
	# 哈希码相同的对象，集合还不能判定该元素是否重复，因为有可能是哈希碰撞（冲突）
    def __hash__(self):
        return hash(self.name)
        
	# 如果两个对象的哈希码相同，集合会使用__eq__判定两个对象是否相同
    # 如果该方法返回True，集合就会认为两个对象是相同的，重复元素不会添加到集合中
    def __eq__(self, other):
        return self.name == other.name and self.age == other.age


p1 = Person('勤勤', 20)
p2 = Person('勤勤', 20)
p3 = Person('勤勤', 18)
persons = {p1, p2, p3}
print(len(persons))
print(persons)

# 2
# {勤勤: 20, 勤勤: 18}

由上可以看出，集合将姓名和年龄相同的对象视为相同对象。

如果想将姓名相同的对象就视为相同对象，那么在写__eq__相等性判断的时候，只写上名字是否相等即可。