__eq__ 方法将两个自定义对象与列表进行比较

Posted 2023-02-22

【中文标题】__eq__ 方法将两个自定义对象与列表进行比较

【英文标题】：__eq__ method to compare two custom objects with list

【发布时间】：2022-01-05 08:42:05

【问题描述】：

我有一个包含字符串列表的 python 类。我想比较这个类的两个实例。但是，我注意到如果我的班级变大，那么比较写起来会很麻烦。有没有比这更简单的方法来编写__eq__ 方法？我使用 List 而不是 Set，因为我想在稍后阶段用来自 JSON 的数据填充这个类。

json_one = "var_one": "hello", "var_two": ["foo", "bar"]
json_two = "var_one": "hello", "var_two": ["bar", "foo"]

@dataclass
class Test:
    var_one: str
    var_two: List[str]

    def __eq__(self, other):
        if isinstance(other, Test):
            return self.var_one == other.var_one\
                   and sorted(self.var_two) == sorted(other.var_two)
        return False


test_one = Test(**json_one)
test_two = Test(**json_two)

if test_one == test_two:
    print("yay we're equal with lists in different order")

【问题讨论】：

并非如此。如果数据类只是属性的相等比较，则数据类可以生成方法，如果您想要比这更复杂的东西，这取决于您。如果var_two的顺序无关紧要，也许一套更合适？

对不起，我不知道这里到底应该是什么“麻烦”。对我来说看起来非常简单和紧凑。我不明白你想在这里简化什么。没有必要进一步简化。 ---- 你应该改变的只有一件事：显然顺序并不重要，你应该为var_two 使用一组，而不是常规列表。

@RegisMay 很好地描绘了一个有 20 个变量的类。这将是一个很大的 if 语句。 List 的原因是我从 JSON 获取数据，并且在该 JSON 数据中有列表。

重点是不要如果默认结构不符合您的需要，只需将 JSON 扩展为 Python 对象。

定义你的类独立的JSON，并添加一个可以专门解析JSON的（类）方法来创建类的实例。

【参考方案1】：

您可以使用json.load 的输出定义您的类的实例这一事实不应决定您的类的设计。而是“自然地”定义您的类，并提供一种使用 JSON 响应创建实例的方法。

from dataclasses import dataclass
import json

@dataclass
class Test:
    # Emphasize the independence of your class from any particular JSON object
    str1: str
    set_of_str1: set[str]

    @classmethod
    def from_json(cls, s):
        d = json.loads(s)
        return cls(d['var_one'], set(d['var_two']))


assert Test("foo", set(["bar", "baz"])) == \
        Test.from_json('"var_one": "foo", "var_two": ["baz", "bar"]')