python - 返回 Cython 数组

Question

如何正确初始化并返回一个 Cython 数组？例如:

cdef public double* cyTest(double[] input):
  cdef double output[3]

  for i in xrange(3):
    output[i] = input[i]**2
    print 'loop: ' + str(output[i])
  return output

cdef double* test = [1,2,3]
cdef double* results = cyTest(test)

for i in xrange(3):
  print 'return: ' + str(results[i])

返回:

loop: 1.0->1.0
loop: 2.0->4.0
loop: 3.0->9.0
return: 1.88706086937e-299
return: 9.7051011575e+236
return: 1.88706086795e-299

很明显，results 仍然只指向垃圾，而不是它应该指向的值。诚然，我对混合使用指针和数组语法以及在 Cython 上下文中哪个更可取/可能感到有点困惑。

最后，我想从纯 C++ 函数调用 cyTest:

#include <iostream>
#include <Python.h>
#include "cyTest.h"

void main() {
  Py_Initialize();
  initcyTest();
  double input[3] = {1,2,3};
  double* output = cyTest(input);

  for(int i = 0; i < 3; i++)
    std::cout << "cout: " << output[i] << std::endl;

  Py_Finalize();
}

这会返回类似的结果:

loop: 1.0->1.0
loop: 2.0->4.0
loop: 3.0->9.0
cout: 1
cout: 6.30058e+077
cout: 6.39301e-308

有人愿意解释我犯了什么错误吗？我想让它尽可能简单。毕竟它只是将一个数组从 Cython 返回到 C++。如果不需要，我稍后会处理动态内存分配。

Answer 1

您正在返回对本地数组 ( output ) 的引用，这将不起作用。

尝试将脚本更改为:

from cpython.mem cimport PyMem_Malloc

cdef public double * cyTest(double[] input):
    cdef double * output = < double * >PyMem_Malloc( sizeof(double) * 3 )
    for i in xrange(3):
        output[i] = input[i]**2
        print 'loop: ' + str(output[i])
    return output

在你的 C++ 代码中，

在你使用完double* output 问题之后 free( output );

如果你想在你的 pyx 脚本中使用 cdef double* results = cyTest(test) 然后不要忘记使用 PyMem_Free(results)