将ndarray从Python-extension返回到C

#include <Python.h>
#include <numpy/arrayobject.h>

// C语言函数，接收ndarray对象作为参数
void process_array(PyObject* array) {
    PyArrayObject* np_array = (PyArrayObject*)array;

    // 获取ndarray的维度和数据类型
    int ndim = PyArray_NDIM(np_array);
    npy_intp* shape = PyArray_SHAPE(np_array);
    int dtype = PyArray_TYPE(np_array);

    // 获取ndarray的数据指针
    void* data = PyArray_DATA(np_array);

    // 在这里进行对ndarray的处理
    // ...

    // 打印ndarray的信息
    printf("ndim: %d\n", ndim);
    printf("shape: ");
    for (int i = 0; i < ndim; i++) {
        printf("%ld ", shape[i]);
    }
    printf("\n");
    printf("dtype: %d\n", dtype);
}

// Python扩展模块的函数，接收Python对象作为参数
static PyObject* process_array_wrapper(PyObject* self, PyObject* args) {
    PyObject* array;

    // 解析Python函数参数
    if (!PyArg_ParseTuple(args, "O", &array)) {
        return NULL;
    }

    // 调用C语言函数，传递ndarray对象
    process_array(array);

    Py_RETURN_NONE;
}

// 模块方法定义
static PyMethodDef module_methods[] = {
    {"process_array", process_array_wrapper, METH_VARARGS, "Process ndarray from Python-extension to C"},
    {NULL, NULL, 0, NULL}
};

// 模块定义
static struct PyModuleDef module_def = {
    PyModuleDef_HEAD_INIT,
    "my_module",
    NULL,
    -1,
    module_methods
};

// 模块初始化函数
PyMODINIT_FUNC PyInit_my_module(void) {
    import_array();  // 初始化NumPy数组对象

    return PyModule_Create(&module_def);
}