重写dataclass的__eq__与__hash__后集合交集结果异常问题

核心问题：__eq__与__hash__的一致性违反 #

Python的哈希表结构（包括set、dict等）依赖一个关键原则：如果两个对象a == b返回True，那么它们的hash(a)必须等于hash(b)。

你的代码中：

• ten（比如digit=10）和fifteen（digit=15）通过自定义__eq__判定为相等，但它们的__hash__返回的是各自digit的哈希值（hash(10)和hash(15)），完全不同。
• 集合在存储元素时，会先根据哈希值将元素分配到不同的哈希桶；查找或计算交集时，只会在相同哈希桶的元素中比较==。因为这两个实例哈希值不同，集合会把它们当成完全无关的元素，自然交集为空。

修复方案：让__eq__与__hash__逻辑一致 #

要解决这个问题，你需要让所有被__eq__判定为相等的对象，返回相同的哈希值。比如，按照你“差值小于10即相等”的逻辑，可以将digit按10为区间取整，用这个分组值来计算哈希：

from dataclasses import dataclass
import numpy as np

@dataclass
class Number:
    digit: int

    def __eq__(self, other):
        # 先判断类型，避免与非Number对象比较出错
        if not isinstance(other, Number):
            return NotImplemented
        return np.abs(self.digit - other.digit) < 10
    
    def __hash__(self):
        # 同一区间内的digit返回相同哈希值
        return hash(self.digit // 10)

测试验证：

• ten = Number(10)，fifteen = Number(15)，ten == fifteen返回True
• hash(ten)和hash(fifteen)均为hash(1)，放入集合后会被分配到同一哈希桶
• 此时set([ten]).intersection(set([fifteen]))会返回包含该实例的集合，符合预期

额外注意事项 #

• 自定义相等逻辑可能打破相等传递性：比如Number(5) == Number(12)、Number(12) == Number(20)，但Number(5) != Number(20)。这种非传递的相等关系在某些场景下可能引发逻辑混乱，使用前需确认业务场景是否允许。
• 始终在__eq__中先判断对方类型，返回NotImplemented而非直接报错，这是Python的规范做法。

内容的提问来源于stack exchange，提问作者Jailbone