简化数据结构的初始化方法

最新推荐文章于 2023-10-04 09:58:14 发布

Lzs1998

最新推荐文章于 2023-10-04 09:58:14 发布

阅读量589

点赞数 1

分类专栏： Python 文章标签： python 数据结构优化

本文链接：https://blog.csdn.net/Lzs1998/article/details/89644492

版权

Python 专栏收录该内容

48 篇文章 3 订阅

订阅专栏

你写了很多仅仅用作数据结构的类，不想写太多烦人的 __init__() 函数。

可以在一个基类中写一个公用的 __init__() 函数，然后使你的类继承自这个基类。

# 公共结构
class Struct:
    _filed = []

    def __init__(self, *args):
        if len(args) != len(self._filed):
            raise TypeError('Expected {} arguments'.format(len(self._filed)))
        for name, value in zip(self._filed, args):
            setattr(self, name, value)


class Stock(Struct):
    _filed = ['name', 'shares', 'price']


class Point(Struct):
    _filed = ['x', 'y']


class Circle(Struct):
    _filed = ['radius']

    def area(self):
        return math.pi * self.radius ** 2


s = Stock('ACME', 50, 91.1)
p = Point(2, 3)
c = Circle(4.5)

当你需要使用大量很小的数据结构类的时候，相比手工一个个定义 __init__() 方法而已，使用这种方式可以大大简化代码。