0

C++ 関数から Python 関数に行列を返したい。配列を返す例であるこのソリューションを確認しました。

10x10例として、で満たされた配列を返したいとし10ます。

関数.cpp:

extern "C" int* function(){
int** information = new int*[10];
for(int k=0;k<10;k++) {
    information[k] = new int[10];
    }
for(int k=0;k<10;k++) {
    for(int l=0;l<10;l++) {
        information[k][l] = 10;
        }
}
return *information;
}

Python コードは次のとおりです 。

import ctypes
from numpy.ctypeslib import ndpointer

lib = ctypes.CDLL('./library.so')
lib.function.restype = ndpointer(dtype=ctypes.c_int, shape=(10,))

res = lib.function()
print res

これをコンパイルするには、次を使用します。

g++ -c -fPIC function.cpp -o function.o
g++ -shared -Wl,-soname,library.so -o library.so function.o

sonameうまくいかない場合は、次を使用しますinstall_name。

g++ -c -fPIC function.cpp -o function.o
g++ -shared -Wl,-install_name,library.so -o library.so function.o

Python プログラムを実行した後python wrapper.pyの出力は次のとおりです。
[10 10 10 10 10 10 10 10 10 10]

10 要素の 1 行のみ。10x10 のマトリックスが欲しいです。私が間違っていることは何ですか？前もって感謝します。

4

1 に答える 1

2

function.cpp:

extern "C" int* function(){
    int* result = new int[100];
    for(int k=0;k<100;k++) {
        result[k] = 10;
    }
    return result;
}

のwrapper.py

lib.function.restype = ndpointer(dtype=ctypes.c_int, shape=(10,)) //incorrect shape
lib.function.restype = ndpointer(dtype=ctypes.c_int, ndim=2, shape=(10,10)) // Should be two-dimensional
于 2013-10-01T09:59:20.600 に答える