python笔记：dataclass

1 引子：其他类似实现方法的局限性

假设我们现在需要实现这样的内容：

name	China
area	960
population	140967

1.1 tuple/list

country1_tuple=('China','960','140967')
country1_tuple[0]
#'China'

缺点：需要记住各个属性是list/tuple第几位的属性（基于位置）

1.2 字典

country1_dict={'name':'China',
              'area':960,
              'population':140967}
country1_dict['name']
#'China'

不足
- 不能用country1_dict.name的形式进行控制
- 少值【country1_dict={'name':'China', 'area':960】一样可以创建成功

1.3 namedtuple

python 库整理： collections.namedtuple_python collections.namedtuple创建一个-CSDN博客

from collections import namedtuple

Country=namedtuple('Country1',['name','area','population'])

country1=Country('China',960,'140967')
country2=Country('Singapore',733,564)

country1
#Country1(name='China', area=960, population='140967')

country2.name
#'Singapore'

弥补了字典的两个缺点
- 可以用country2.name的形式
- 少值会报错
不足
- 无法修改值

1.4 自定义类

class Country:
    def __init__(self,name,area,population):
        self.name=name
        self.area=area
        self.population=population
        
country1_class=Country('China',960,140967)
country1_class.name
#'China'


country1_class.name="PRC"

解决了namedtuple无法修改值的问题
不足：
- __init__方法中重复代码 (示例中每个属性都需要写3遍)

2 数据类dataclass

提供了一个简写__init__方法的语法糖.
类型注释是必填项
默认值的传递方式和__init__方法的参数格式一致

from dataclasses import dataclass

@dataclass
class Country_dc:
    name:str
    area:int
    population:int=40
        
country_dc1=Country_dc('China',960,140967)
country_dc2=Country_dc('Singapore',733)


country_dc1
#Country_dc(name='China', area=960, population=140967)


country_dc2
#Country_dc(name='Singapore', area=733, population=40)

2.1 数据嵌套

from typing import List

@dataclass
class Countrys_dc:
    name:str
    countries:List[Country_dc]
country_lst=Countrys_dc('China & SG',[country_dc1,country_dc2])
country_lst
'''
Countrys_dc(name='China & SG', countries=[Country_dc(name='China', area=960, population=140967), Country_dc(name='Singapore', area=733, population=40)])
'''