如何在python中創建自定義列表類？

Question

我想用Python語言編寫自定義列表類（姑且稱之爲MyCollection），在那裏我可以最終調用：

for x in myCollectionInstance: 
    #do something here 

我怎麼會去這樣做呢？是否有一些課程需要延續，或者是否有我必須覆蓋的功能？

Answer 1

Python Language Reference, §3.4.6, "Emulating container types"

Answer 2

你可以繼承list如果您的收藏基本上就像一個列表：

class MyCollection(list): 
    def __init__(self, *args, **kwargs): 
     super(MyCollection, self).__init__(args[0]) 

但是，如果你的主要的心願就是您的收藏支持迭代器協議，你只需要提供__iter__方法：

class MyCollection(object): 
    def __init__(self): 
     self._data = [4, 8, 15, 16, 23, 42] 

    def __iter__(self): 
     for elem in self._data: 
      yield elem 

這可以讓你的任何實例遍歷。

Answer 3

你可以擴展list類：

class MyList(list): 

    def __init__(self, *args): 
     super(MyList, self).__init__(args[0]) 
     # Do something with the other args (and potentially kwars) 

用法示例：

a = MyList((1,2,3), 35, 22) 
print(a) 
for x in a: 
    print(x) 

預期輸出：

[1, 2, 3] 
1 
2 
3 

Answer 4

我想繼承MutableSequence，作爲recommended by Alex Martelli。這適用於我，特別是當我需要在我創建的列表頂部添加自定義方法時。

from collections import MutableSequence 

class MyList(MutableSequence): 
    """A container for manipulating lists of hosts""" 
    def __init__(self, data=None): 
     """Initialize the class""" 
     super(MyList, self).__init__() 
     if (data is not None): 
      self._list = list(data) 
     else: 
      self._list = list() 

    def __repr__(self): 
     return "<{0} {1}>".format(self.__class__.__name__, self._list) 

    def __len__(self): 
     """List length""" 
     return len(self._list) 

    def __getitem__(self, ii): 
     """Get a list item""" 
     return self._list[ii] 

    def __delitem__(self, ii): 
     """Delete an item""" 
     del self._list[ii] 

    def __setitem__(self, ii, val): 
     # optional: self._acl_check(val) 
     self._list[ii] = val 

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

    def insert(self, ii, val): 
     # optional: self._acl_check(val) 
     self._list.insert(ii, val) 

    def append(self, val): 
     self.insert(len(self._list), val) 

if __name__=='__main__': 
    foo = MyList([1,2,3,4,5]) 
    foo.append(6) 
    print foo # <MyList [1, 2, 3, 4, 5, 6]>