2009-05-28 31 views
6

這是對問題912526 - How do I pass lots of variables to and from a function in Python?的後續處理。如何使用派生自Python列表類的類

在我編寫的程序中需要傳遞很多變量,從我之前的問題中我明白我應該將這些變量放入類中,然後傳遞類。

其中一些變量來自重複設置 - 對於需要跟蹤多個層的光學屬性(折射率,吸收率,厚度等)的薄膜計算。

是否存儲像這樣的變量以創建一個派生自Python列表的類的最好方法來存儲每個容納單個圖層變量的類集?然後將處理層中的一組圖層的函數放置在派生自列表的類中,以及處理該類中特定圖層的函數?有沒有更好的方法來做到這一點與單個類?

使用下面的例子中的兩個班的做法,我能夠做一些事情,讓我可以訪問使用statments變量,如

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

你真的需要從`list`繼承了`Layers`類?只需使用列表。 – 2009-05-28 18:34:22

回答

9

有幾個問題在你的代碼:

1.如果你做任何列表操作的結果將是一個原生列表:

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

2.爲什麼定義input_string,當你可以覆蓋__repr____str__

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 
+0

@hopatong:你可以通過實現__add__方法來解決第一個問題。請參閱http://docs.python.org/reference/datamodel.html#emulating-numeric-types。 – 2009-05-28 18:29:40

相關問題