Python中的dataclass：简化数据类的创建 - 文章 - 开发者社区

Python中的dataclass是一个装饰器，用于自动添加一些常见的方法，如构造函数、__repr__、__eq__等。它简化了创建数据类的过程，减少了样板代码，提高了代码的可读性和可维护性。有点类似java里面的Java Bean。

让我们看一个简单的例子来说明dataclass的用法：

  
from dataclasses import dataclass  
  
  
@dataclass  
class User:  
 name: str  
 age: int  
  
  
p1 = User("小博", 18)  
p2 = User("小博", 18)  
p3 = User("小明", 19)  
  
print(p1) # Output: User(name='小博', age=18)  
print(p1.__repr__()) # Output: User(name='小博', age=18)  
print(p1 == p2) # Output: True  
print(p1.__eq__(p2)) # Output: True  
print(p1 == p3) # Output: False

在上面的例子中，我们定义了一个名为User的数据类，它有两个成员变量：name和age。在这个简单的例子中，dataclass自动为我们创建了以下方法：

__init__: 自动添加了带有name和age参数的构造函数，我们可以用User("小博", 18)的形式创建对象。
__repr__: 自动添加了一个友好的表示对象的字符串方法，我们可以通过print()函数查看对象的内容。
__eq__: 自动添加了对象之间的相等比较方法，我们可以使用==来比较两个对象是否相等。

接下来让我们看一下dataclass的对象之间怎么比较

  
from dataclasses import dataclass, field  
  
  
@dataclass(order=True)  
class Person:  
 name: str  
 age: int = field(compare=False) # 指定某个字段不参与排序  
 height: float  
  
  
# 创建实例  
person1 = Person("Alice", 30, 185.5)  
person2 = Person("Bob", 25, 170.0)  
person3 = Person("Charlie", 35, 175.2)  
person4 = Person("Alice", 35, 175.2)  
person5 = Person("Bob", 25, 185.2)  
  
# 使用比较运算符进行排序  
people = [person1, person2, person3, person4, person5]  
people_sorted = sorted(people) # 默认升序  
people_sorted = sorted(people, reverse=True)  
  
for person in people_sorted:  
 print(person.name, person.age, person.height)  
  
输出结果：  
Charlie 35 175.2  
Bob 25 185.2  
Bob 25 170.0  
Alice 30 185.5  
Alice 35 175.2

要让类的实例对象之间进行比较，需要在定义类的时候，加上@dataclass(order = True)。默认将生成__gt__、__ge__、__lt__、__le__方法。如果传入False，则省略它们。默认会按照类中定义的字段顺序进行对比，第一个字段的值相等的时候，就用第二个字段进行比较。要忽略某个字段不进行对比的话，可以使用field(compare=False)

End

如果觉得文章对你有帮助的话，欢迎点赞转发~

如果还没有进粉丝交流群的小伙伴，赶快添加好友(xiaobotester)邀请你们进群喔。

👇👇👇

关注公众号，测试干货及时送达 picture.image