如何使用从Python的列表类派生的类

时间:2009-05-28 15:05:14

标签: python class

这是问题912526的后续问题 - How do I pass lots of variables to and from a function in Python?

在我正在编写的程序中需要传递很多变量,而且根据我之前的问题,我理解我应该将这些变量放入类中,然后传递类。

其中一些变量来自重复集 - 对于薄膜计算,我需要跟踪多个层的光学特性(折射率,吸收,厚度等)。

存储这样的变量的最佳方法是创建一个派生自Python列表的类来存储每组保存单个层的变量的类吗?然后将处理从列表中派生的图层集的函数和处理该类中特定图层的函数放在一起?有没有更好的方法来完成单一课程?

在以下示例中使用两类方法,我可以进行设置,以便可以使用

等语句访问变量
n1 = layers[5].n

这是最好的方法,对吗?

#Test passing values to and from functions

class Layers(list):

    def add(self,n,k,comment):
        self.append( OneLayer(n,k,comment) )

    def input_string(self):
        input_string = []
        for layer in self:
            vars = layer.input_string()
            for var in vars:
                input_string.append( var )
        return input_string

    def set_layers(self,results):
        for layer,i in enumerate(self):
            j = i*layer.num_var
            layer.set_layer( *results[j:j+2] )

class OneLayer(object):

    def __init__(self,n,k,comment):
        self.n = n
        self.k = k
        self.comment = comment

    def input_string(self):
        return [['f','Index of Refraction',self.n], ['f','Absorption',self.k],['s','Comment',self.comment]]

    def set_layer(self,n,k,comment):
        self.n = n; self.k=k; self.comment = comment

    def num_var(self):
        return 3

if __name__ == '__main__':
    layers = Layers()

    layers.add(1.0,0.0,'This vacuum sucks')
    layers.add(1.5,0.0,'BK 7 Glass')

    print layers[0].n
    print layers.input_string()
    layers[1].set_layer(1.77,0.0,'Sapphire')
    print layers.input_string()

我从这个测试程序得到以下输出:

1.0
[['f', 'Index of Refraction', 1.0], ['f', 'Absorption', 0.0], ['s', 'Comment', 'This vacuum sucks'], ['f', 'Index of Refraction', 1.5], ['f', 'Absorption', 0.0], ['s', 'Comment', 'BK 7 Glass']]
[['f', 'Index of Refraction', 1.0], ['f', 'Absorption', 0.0], ['s', 'Comment', 'This vacuum sucks'], ['f', 'Index of Refraction', 1.77], ['f', 'Absorption', 0.0], ['s', 'Comment', 'Sapphire']]

1 个答案:

答案 0 :(得分:9)

您的代码中存在以下几个问题:

1.如果您进行任何列表操作,结果将是本机列表:

layers1 = Layers()
layers2 = Layers()
layers1 + layers2 -> the result will be a native list

2.为什么在覆盖__repr____str__

时定义input_string

3.为什么你甚至不得不在这种情况下从列表中得出?如果您希望您的类的行为与列表完全相同,则只需从列表派生。但在你的情况下,你似乎在寻找一个容器。 要使类的行为类似于列表,您需要做的就是覆盖一些特殊的python方法http://docs.python.org/reference/datamodel.html#emulating-container-types

class Layers(object):
    def __init__(self, container=None):
        if container is None:
            container = []
        self.container = container

    def add(self,n,k,comment):
        self.container.append([n,k,comment])

    def __str__(self):
        return str(self.container)

    def __repr__(self):
        return str(self.container)

    def __getitem__(self, key):
        return Layers(self.container[key])

    def __len__(self):
        return len(self.container)

>>> l = Layers()
>>> l.add(1, 2, 'test')
>>> l.add(1, 2, 'test')
>>> l
[[1, 2, 'test'], [1, 2, 'test']]
>>> l[0]
[1, 2, 'test']
>>> len(l)
2