如何通过Python脚本在两个Cython模块间传递指针变量？——Cython类指针传递的实现细节问询

我来帮你解决这个跨Cython模块传递指针的问题，分两种方案，一种是通过内存地址（整数）传递，另一种是更安全的直接持有对象引用的方式，后者更推荐哦。

方案1：通过内存地址（整数）传递指针 #

这种方式契合你原本设想的Python层传递指针的场景，但要重点注意内存安全问题。

步骤1：修改A.pyx，实现get_ptr()方法 #

Python无法直接处理C指针类型，所以我们需要把C指针转换成Python能识别的整数（即内存地址）返回：

# A.pyx
from libc.stdlib cimport malloc, free

cdef class A:
    cdef float * ptr
    
    def __init__(self):
        # 初始化时分配内存，避免野指针
        self.ptr = <float*>malloc(sizeof(float))
        if self.ptr is NULL:
            raise MemoryError("Failed to allocate memory for A's ptr")
    
    cpdef get_ptr(self):
        # 将C指针转换为Python整数（内存地址）
        return <size_t>self.ptr
    
    cpdef set_value(self, float val):
        # 方便测试的辅助方法，给指针指向的内存赋值
        if self.ptr is not NULL:
            self.ptr[0] = val
    
    def __dealloc__(self):
        # 析构时释放内存，防止内存泄漏
        if self.ptr is not NULL:
            free(self.ptr)

步骤2：修改B.pyx，实现submit方法 #

在B类中，把传递过来的整数转换回C的float*指针：

# B.pyx
cdef class B:
    cdef float * f_ptr
    
    cpdef submit(self, ptr_var):
        # 将Python整数转换回float*指针
        self.f_ptr = <float*>ptr_var
    
    cpdef float get_value(self):
        # 访问指针指向的数据
        if self.f_ptr is not NULL:
            return self.f_ptr[0]
        raise ValueError("Pointer is invalid or not set")

Python调用代码 #

from A import A
from B import B

a = A()
a.set_value(3.14)  # 给指针指向的内存赋值
b = B()
ptr_addr = a.get_ptr()
b.submit(ptr_addr)
print(b.get_value())  # 输出: 3.14

⚠️ 内存安全提醒：如果a对象被Python垃圾回收（比如执行del a或者对象超出作用域），A的__dealloc__方法会释放ptr的内存，此时b.f_ptr就变成了野指针，访问会导致程序崩溃。所以必须保证a的生命周期长于b。

方案2：直接持有A对象的引用（更安全，推荐使用） #

这种方式不需要手动传递指针，让B持有A的引用，直接访问A的cdef成员指针，既安全又简洁，还能彻底避免野指针问题。

步骤1：修改A.pyx（无需get_ptr方法） #

# A.pyx
from libc.stdlib cimport malloc, free

cdef class A:
    cdef float * ptr
    
    def __init__(self):
        self.ptr = <float*>malloc(sizeof(float))
        if self.ptr is NULL:
            raise MemoryError("Failed to allocate memory for A's ptr")
    
    cpdef set_value(self, float val):
        if self.ptr is not NULL:
            self.ptr[0] = val
    
    def __dealloc__(self):
        if self.ptr is not NULL:
            free(self.ptr)

步骤2：修改B.pyx，cimport A类并直接访问指针 #

在B.pyx开头先cimport A类，这样就能直接访问A的cdef成员ptr：

# B.pyx
# 确保A已经被编译，或在setup.py中把A和B放在一起编译
from A cimport A

cdef class B:
    cdef float * f_ptr
    cdef object a_reference  # 持有A对象的引用，防止被GC回收
    
    cpdef submit(self, A a_obj):
        self.a_reference = a_obj  # 保存引用，避免A被提前回收
        self.f_ptr = a_obj.ptr    # 直接获取A的cdef指针
    
    cpdef float get_value(self):
        if self.f_ptr is not NULL:
            return self.f_ptr[0]
        raise ValueError("Pointer is invalid or not set")

Python调用代码 #

from A import A
from B import B

a = A()
a.set_value(3.14)
b = B()
b.submit(a)  # 直接传入A对象，不需要手动传指针
print(b.get_value())  # 输出: 3.14

这种方式的核心优势是：只要b对象存在，a就不会被垃圾回收，指针始终有效，完全规避了野指针风险。

编译注意事项 #

不管用哪种方案，都需要通过setup.py编译Cython模块，示例setup.py如下：

from setuptools import setup
from Cython.Build import cythonize

setup(
    ext_modules=cythonize(["A.pyx", "B.pyx"]),
)

编译命令：python setup.py build_ext --inplace

内容的提问来源于stack exchange，提问作者Claas Bontus