Cython: C++ Используйте вектор в словаре?

Question

Я использую следующий код, чтобы попытаться работать с векторами C++:

from libcpp.vector cimport vector                                   

cdef struct StartEnd: 
    long start, end 

cdef vector[StartEnd] vect 
print(type(vect)) 
cdef int i 
cdef StartEnd j 
k = {} 
k['hi'] = vect 
for i in range(10): 
    j.start = i 
    j.end = i + 2 
    k['hi'].push_back(j) 
for i in range(10): 
    print(k['hi'][i]) 

Точная функциональность здесь не имеет значения, это просто манекен программы. Проблема заключается в том, что при запуске это порождает ошибку: AttributeError: 'list' object has no attribute 'push_back' Это работает, если нет словаря, но я считаю, что словарь необходим для моего использования. Есть ли способ сделать эту работу?

Я не хочу копировать векторы взад и вперед, поскольку эти векторы получат десятки миллионов записей. Может быть, я могу вместо этого сохранить указатели на вектор?

Answer 1

Вектор C++ автоматически преобразуется в list на границе Cython/Python (следовательно, появляется сообщение об ошибке). Python dict ожидает хранения объектов Python, а не векторов C++. Создание cdef class, который содержит C++ вектор и положить, что в Словаре вместо:

cdef class VecHolder: 
    cdef vector[StartEnd] wrapped_vector 

    # the easiest thing to do is add short wrappers for the methods you need 
    def push_back(self,obj): 
    self.wrapped_vector.push_back(obj) 

cdef int i 
cdef StartEnd j 
k = {} 
k['hi'] = VecHolder() 
for i in range(10): 
    j.start = i 
    j.end = i + 2 
    k['hi'].push_back(j) # note that you're calling 
     # the wrapper method here which then calls the c++ function