[已解决]有那个大佬用C写过python扩展，有一些问题请教

Stubborn

想用C语言给python写一个扩展

简单的例子一：
我在python中，使用我的扩展函数，可以正常，打印得到13；

1. from my_method import fir_test_add
   
2. 
   
3. print(fir_test_add(8, 5))

复制代码

对应的C原形是

1. int test_add(int a, int b)
   
2. {
   
3.     return a + b ;
   
4. }
   
5. 
   
6. static PyObject  *
   
7. Test_add(PyObject *self, PyObject *args){
   
8.     int a,b;
   
9.     if(!PyArg_ParseTuple(args ,"ii", &a, &b)) {
   
10.         return NULL;
    
11.     }
    
12.     return (PyObject*)Py_BuildValue("i", test_add(a, b));
    
13. }

复制代码

问题来了，如果我的python输入是一个列表长度不定，希望返回的也是一个固定长度的列表。

1. from fir_method import fir_test_add
   
2. 
   
3. print(fir_test_add([8, 5, 8, 9, 1, 8]))

复制代码

那么我对应的C接口应该怎么编写？

最佳答案

月排行榜 / 总排行榜

Stubborn

2022-12-28 02:10:46

hrpzcf 发表于 2022-12-27 23:17
研究过文档，奈何 C 基础不够扎实看不明白

谢谢，看完了

1. static PyObject *
   
2. Update(PyObject *self, PyObject *args)
   
3. {
   
4.     PyObject *PyList;
   
5.     PyObject *PyItem;
   
6.     Py_ssize_t length;
   
7. 
   
8.     if(!PyArg_ParseTuple(args ,"O!", &PyList_Type, &PyList)) {
   
9.         PyErr_SetString(PyExc_TypeError, "params must be a List");
   
10.         return NULL;
    
11.     }
    
12. 
    
13.     length = PyList_Size(PyList);
    
14.     int array[length];
    
15.     for (int idx = 0; idx < length; ++idx)
    
16.     {
    
17.         PyItem = PyList_GetItem(PyList, idx);
    
18.         array[idx] = PyLong_AsLong(PyItem);
    
19.     }
    
20.     int *arr = update_state(array, length);
    
21.     int val0, val1, val2, val3, val4, val5, val6, val7, val8, val9;
    
22.     val0 = arr[0];
    
23.     val1 = arr[1];
    
24.     val2 = arr[2];
    
25.     val3 = arr[3];
    
26.     val4 = arr[4];
    
27.     val5 = arr[5];
    
28.     val6 = arr[6];
    
29.     val7 = arr[7];
    
30.     val8 = arr[8];
    
31.     val9 = arr[9];
    
32.     return (PyObject*)Py_BuildValue("[i,i,i,i,i,i,i,i,i,i]", val0, val1, val2, val3, val4, val5, val6, val7, val8, val9);
    
33. }

复制代码

跳转到最佳答案楼层

hrpzcf

研究过文档，奈何 C 基础不够扎实看不明白

1. https://docs.python.org/zh-cn/3.7/c-api/list.html
   
2. https://docs.python.org/zh-cn/3.7/c-api/allocation.html#c.PyObject_New
   
3. https://docs.python.org/zh-cn/3.7/c-api/memory.html#object-allocators

复制代码

Stubborn

hrpzcf 发表于 2022-12-27 23:17
研究过文档，奈何 C 基础不够扎实看不明白

谢谢，看完了

1. static PyObject *
   
2. Update(PyObject *self, PyObject *args)
   
3. {
   
4.     PyObject *PyList;
   
5.     PyObject *PyItem;
   
6.     Py_ssize_t length;
   
7. 
   
8.     if(!PyArg_ParseTuple(args ,"O!", &PyList_Type, &PyList)) {
   
9.         PyErr_SetString(PyExc_TypeError, "params must be a List");
   
10.         return NULL;
    
11.     }
    
12. 
    
13.     length = PyList_Size(PyList);
    
14.     int array[length];
    
15.     for (int idx = 0; idx < length; ++idx)
    
16.     {
    
17.         PyItem = PyList_GetItem(PyList, idx);
    
18.         array[idx] = PyLong_AsLong(PyItem);
    
19.     }
    
20.     int *arr = update_state(array, length);
    
21.     int val0, val1, val2, val3, val4, val5, val6, val7, val8, val9;
    
22.     val0 = arr[0];
    
23.     val1 = arr[1];
    
24.     val2 = arr[2];
    
25.     val3 = arr[3];
    
26.     val4 = arr[4];
    
27.     val5 = arr[5];
    
28.     val6 = arr[6];
    
29.     val7 = arr[7];
    
30.     val8 = arr[8];
    
31.     val9 = arr[9];
    
32.     return (PyObject*)Py_BuildValue("[i,i,i,i,i,i,i,i,i,i]", val0, val1, val2, val3, val4, val5, val6, val7, val8, val9);
    
33. }

复制代码