c – 如何使用swig Python重载下标运算符

我有一个包含std :: vector< Foo>的类.其中Foo是一个包含键,值,注释等的类.请注意,我使用向量而不是字典是有原因的.

我在C中重载了下标运算符,使得foos [“Key Name”]将在向量中搜索具有键匹配“Key Name”的Foo对象(其中foos是std :: vector< Foo>).

我使用SWIG为我的库创建一个Python包装器,我真的希望这个下标操作符扩展到Python.换句话说,我希望能够使用foos [“Key Name”]在Python中查找向量中的对象.

有关如何使SWIG识别下标运算符并在Python中重载它的任何提示?我有点惊讶,我找不到人们在网上这样做的例子.我猜大多数人只是使用std :: map并让SWIG将其转换为Python dict.

最佳答案 在直接Python中,如果要重载下标运算符,则应创建__getitem__和__setitem__类方法.举个简单的例子:

class MyClass(object):
    def __init__(self):
        self.storage = {}

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

    def __setitem__(self, key, value):
        self.storage[key] = value

所以,如果你想让C处理这个,我最好的猜测(不,我还没有证实这一点)就是你要在C中创建一个__getitem__和__setitem__.您可以直接在C代码中执行此操作,也可以使用SWIG中的%extend指令来调用C []运算符.

点赞