使用相同的参数创建数据类的多个单独实例

Question

我有一个简单的 Python class，看起来像这样：

class RegularVariable:
    def __init__(self, num_states):
        self.num_states = num_states

我最近发现了 dataclasses 模块，并将我的简单 class 替换为数据 class，如下所示：

@dataclass(frozen=True)
class DataClassVariable:
    num_states: int

（设置 frozen=True 是因为我确实打算将 class 设为只读。）

现在，在我的代码中的某个时刻，我需要用完全相同的 num_states 创建这个 class 的一堆实例。我可以用旧的、简单的 RegularVariable class:

轻松做到这一点

print(len(set([RegularVariable(1) for _ in range(10)])))
# prints 10 as expected

然而，当我用我的 dataclass 版本尝试同样的事情时：

print(len(set([DataClassVariable(1) for _ in range(10)])))
# prints 1

为什么 dataclass 会这样？鉴于此，我如何为我的 DataClassVariable?

创建 10 个不同的对象（实例化）

Answer 1

根据评论，解决方案是添加eq=False

@dataclass(frozen=True, eq=False)
class DataClassVariable:
    num_states: int

Answer 2

当 frozen=True 时，dataclass 将尝试生成 __hash__ 函数，并且仅基于 num_states.

的值进行生成

您可以将 __hash__ 明确定义为

@dataclass(frozen=True)
class DataClassVariable:
    num_states: int
    def __hash__(self):
        return super().__hash__()

以防止自动生成的函数使每个实例看起来相同。请注意，这不需要禁用 __eq__.

的自动生成

Creating multiple separate instances of a dataclass with the same arguments