使用SWIG從Python傳遞和數組參數到C

Question

我是第一次使用SWIG + Python + C，我在將數組從Python傳遞給C時遇到了問題。

這是C中的函數簽名。

my_setup(char * my_string, int my_count, int my_types[], double my_rate, int my_mode);

我想從Python中調用這個C函數，如下所示

my_array = [1, 2, 3, 4, 5, 6]
my_setup("My string", 6, my_array, 50, 0)

但我不知道如何構造數組my_array 。 我得到的錯誤是

Traceback (most recent call last):
  File "test_script.py", line 9, in <module>
    r = my_library.my_setup("My string", 6, my_types, 50, 0)
TypeError: in method 'my_setup', argument 3 of type 'int []'

我嘗試將SWIG接口文件用於numpy和ctypes ， 但未成功 。

我希望有人可以幫助我傳遞一個數組作為函數my_setup的第三個參數。

另外，這是我的第一個堆棧溢出帖子！

Answer 1

解析my_setup()的Python列表，而不是嘗試在SWIG .i文件中翻譯它。 更改

my_setup(char * my_string, int my_count, int my_types[], double my_rate, int my_mode);

至

my_setup(char * my_string, int my_count, PyObject *int_list, double my_rate, int my_mode);

並在my_setup中

    int *array = NULL;
    if ( PyList_Check( int_list ) )
    {
        int nInts = PyList_Size( int_list );
        array = malloc( nInts * sizeof( *array ) );
        for ( int ii = 0; ii < nInts; ii++ )
        {
            PyObject *oo = PyList_GetItem( int_list, ii );
            if ( PyInt_Check( oo ) )
            {
                array[ ii ] = ( int ) PyInt_AsLong( oo );
            }
        }
    }

您必須添加錯誤檢查。 從C開始，當您使用SWIG時，總是將PyObject *返回給Python。 這樣，您可以使用PyErr_SetString()並返回NULL以拋出異常。