C语言中的Python扩展-来回传递列表

#include <Python.h>

static PyObject* process_list(PyObject* self, PyObject* args) {
    PyObject* input_list;
    if (!PyArg_ParseTuple(args, "O", &input_list)) {
        return NULL;
    }

    if (!PyList_Check(input_list)) {
        PyErr_SetString(PyExc_TypeError, "Expected a list");
        return NULL;
    }

    Py_ssize_t input_size = PyList_Size(input_list);
    PyObject* output_list = PyList_New(input_size);

    for (Py_ssize_t i = 0; i < input_size; i++) {
        PyObject* item = PyList_GetItem(input_list, i);
        // 处理列表中的元素，这里只是简单示例
        PyObject* processed_item = PyLong_FromLong(PyLong_AsLong(item) * 2);
        PyList_SetItem(output_list, i, processed_item);
    }

    return output_list;
}

static PyMethodDef module_methods[] = {
    {"process_list", process_list, METH_VARARGS, "Process a list"},
    {NULL, NULL, 0, NULL}
};

static struct PyModuleDef module_definition = {
    PyModuleDef_HEAD_INIT,
    "myextension",
    "A C extension module",
    -1,
    module_methods
};

PyMODINIT_FUNC PyInit_myextension(void) {
    return PyModule_Create(&module_definition);
}