如何用Cython编写返回mpz_t动态数组的函数？

直接返回mpz_t*裸指针给Python层会引发内存泄漏风险，且Python无法直接管理这类C内存。更稳妥的是采用以下两种可行方案：

方案1：返回Python列表（gmpy2.mpz对象） #

这种方式安全性最高，Python可自动管理内存，实现简单：

cimport cython
from gmpy2 cimport mpz, mpz_t, __gmpz_init, __gmpz_set, __gmpz_clear

cpdef list pows(mpz number):
    cdef mpz_t temp
    cdef list result = []
    cdef mpz py_mpz
    
    __gmpz_init(temp)
    __gmpz_set(temp, number._mpz)  # 从输入的gmpy2.mpz获取底层mpz_t
    
    # 模拟动态扩容逻辑：循环添加元素
    for i in range(5):
        __gmpz_mul_ui(temp, temp, 2)  # 示例操作：当前值乘以2
        # 将mpz_t转为gmpy2.mpz对象加入列表
        py_mpz = mpz.__new__(mpz)
        __gmpz_set(py_mpz._mpz, temp)
        result.append(py_mpz)
    
    __gmpz_clear(temp)
    return result

方案2：返回带内存管理的动态mpz_t数组 #

若必须返回底层mpz_t数组，可封装为Cython结构体，同时手动管理内存：

cimport cython
from libc.stdlib cimport malloc, realloc, free
from gmpy2 cimport mpz, mpz_t, __gmpz_init, __gmpz_set, __gmpz_clear

# 定义结构体保存数组指针与长度，方便统一管理
cdef struct MpzArray:
    mpz_t* data
    int length

# 辅助函数：释放数组内存，避免泄漏
cdef void free_mpz_array(MpzArray arr):
    cdef int i
    for i in range(arr.length):
        __gmpz_clear(&arr.data[i])
    free(arr.data)

cpdef MpzArray pows(mpz number):
    cdef MpzArray arr
    cdef int initial_size = 2
    cdef int current_length = 0
    cdef mpz_t temp
    
    # 初始分配内存
    arr.data = <mpz_t*>malloc(initial_size * sizeof(mpz_t))
    arr.length = initial_size
    
    __gmpz_init(temp)
    __gmpz_set(temp, number._mpz)
    
    # 模拟动态扩容逻辑
    for i in range(5):
        # 容量不足时扩容（翻倍）
        if current_length >= arr.length:
            arr.length *= 2
            arr.data = <mpz_t*>realloc(arr.data, arr.length * sizeof(mpz_t))
        
        # 初始化新mpz_t并赋值
        __gmpz_init(&arr.data[current_length])
        __gmpz_set(&arr.data[current_length], temp)
        current_length += 1
        
        __gmpz_mul_ui(temp, temp, 2)
    
    # 可选：调整数组到实际元素长度
    arr.length = current_length
    arr.data = <mpz_t*>realloc(arr.data, arr.length * sizeof(mpz_t))
    
    __gmpz_clear(temp)
    return arr

关键说明： #

• 方案2中返回的MpzArray必须手动调用free_mpz_array释放内存，否则会造成内存泄漏。
• mpz_t[:]类型的memoryview仅适用于固定大小数组，不支持动态扩容场景。

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