如何在 Python 中将自定义类设为集合

我有 Matlab 背景。在 matlab 中，我可以创建一个类定义，然后创建一个对象数组。我可以轻松地使用索引取消引用每个对象。此外，当我从对象数组（没有索引）调用方法时，我可以访问数组中的所有对象。例如，假设 myNewClass 具有属性 .data 和 .text，它还具有方法 .clob。我可以：

% init
a(1) = myNewClass;
a(2) = myNewClass;

a(1).data = [0;0;0];
a(2).data = [1;1;1];

所以现在，如果我调用 a.clob （不是 a(1).clob 或 a(2).clob），我可以做类似的事情

% this is a function inside the methods definition of my class definition
function clob(self)
   % self here is my object array, "a", from above
   for i=1:length(self)
       % deref a single object
       self(i).goClobYourself;
   end
end

我如何在Python中做这样的事情？请注意，我希望我的类成为一个索引集合，有点像列表。但是，我不希望我的“班级列表”接受任何班级，只接受 myNewClass。如果我继承自“list”，我的类将只是一个具有属性 .data、.text 和函数 .clob 的“list”类吗？另请注意，我不需要包含我的对象的“列表”，我希望我的对象是列表，以便我可以从索引中取消引用它们：a[1].clob() 或 a(1).clob( ） （？？ 或类似的东西）。或者，我想将整个数组传递给 self：a.clob() 使我可以访问该列表。我可能对术语有点模糊。

此致

使用 Python 编程时，执行类型检查并不常见。您是否确定需要您的列表仅接受一种类型（及其子类型），或者您是否相信程序员会阅读您的文档而不会将不应该存在的内容放入其中？

如果是这样，这里有一个继承自的示例泛型类collections.MutableSequence这可能会做你想要的：

from collections import MutableSequence
class VerifierList(MutableSequence):
    _list = None
    def __init__(self, allowedClasses, *args, **kwargs):
        super(VerifierList, self).__init__()
        self._list = list(*args, **kwargs)
        self.allowedClasses = tuple(allowedClasses)
    def __repr__(self):
        return repr(self._list)
    def __str__(self):
        return str(self._list)
    def __len__(self):
        return len(self._list)
    def __getitem__(self, index):
        return self._list[index]
    def __setitem__(self, index, value):
        if not isinstance(value, self.allowedClasses):
            raise TypeError('Value of type %s not allowed!' % value.__class__.__name__)
        self._list[index] = value
    def __delitem__(self, index):
        del self._list[index]
    def insert(self, index, value):
        if not isinstance(value, self.allowedClasses):
            raise TypeError('Value of type %s not allowed!' % value.__class__.__name__)
        self._list.insert(index, value)

使用方法如下：

>>> class A(object): pass

>>> class B(object): pass

>>> l = VerifierList((A,))
>>> l.append(A())
>>> print(l)
>>> [<__main__.A object at 0x000000000311F278>]
>>> l.append(B())

Traceback (most recent call last):
  File "<pyshell#228>", line 1, in <module>
    l.append(B())
  File "C:\Python27\lib\_abcoll.py", line 661, in append
    self.insert(len(self), value)
  File "<pyshell#204>", line 23, in insert
    raise TypeError('Value of type %s not allowed!' % value.__class__.__name__)
TypeError: Value of type B not allowed!